ma_model_classifier.h

Go to the documentation of this file.

#ifndef _MA_MODEL_CLASSIFIER_H
#define _MA_MODEL_CLASSIFIER_H
 
#include <vector>
 
#include "../cv/ma_cv.h"
 
#include "ma_model_base.h"
 
namespace ma::model {
 
class Classifier : public Model {
protected:
    ma_tensor_t input_;
    ma_tensor_t output_;
    ma_img_t img_;
    bool is_nhwc_;
    const ma_img_t* input_img_;
    double threshold_score_;
    std::forward_list<ma_class_t> results_;
 
protected:
    ma_err_t preprocess() override;
    ma_err_t postprocess() override;
 
public:
    Classifier(Engine* engine);
    virtual ~Classifier();
    static bool isValid(Engine* engine);
    const std::forward_list<ma_class_t>& getResults();
    const void *getInput();
    ma_err_t run(const ma_img_t* img);
    ma_err_t setConfig(ma_model_cfg_opt_t opt, ...) override;
    ma_err_t getConfig(ma_model_cfg_opt_t opt, ...) override;
};
 
}  // namespace ma::model
 
#endif  // _MA_MODEL_DETECTOR_H