SSCMA-Micro CPP SDK  v2.0.0
SSCMA-Micro is a cross-platform machine learning inference framework designed for embedded devices.
ma::model::Detector Class Reference

#include <ma_model_detector.h>

Inheritance diagram for ma::model::Detector:
Collaboration diagram for ma::model::Detector:

Public Member Functions

 Detector (Engine *engine, const char *name, ma_model_type_t type)
 
virtual ~Detector ()
 
const std::forward_list< ma_bbox_t > & getResults ()
 
const void * getInput () override
 
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
 
- Public Member Functions inherited from ma::Model
 Model (Engine *engine, const char *name, uint16_t type)
 
virtual ~Model ()
 
const ma_perf_t getPerf () const
 
const char * getName () const
 
ma_model_type_t getType () const
 
ma_input_type_t getInputType () const
 
ma_output_type_t getOutputType () const
 
void setPreprocessDone (std::function< void(void *)> func)
 
void setPostprocessDone (std::function< void(void *)> func)
 
void setRunDone (std::function< void(void *)> func)
 
void setUserCtx (void *ctx)
 

Protected Member Functions

ma_err_t preprocess () override
 
- Protected Member Functions inherited from ma::Model
virtual ma_err_t postprocess ()=0
 
ma_err_t underlyingRun ()
 

Protected Attributes

ma_tensor_t input_
 
ma_img_t img_
 
const ma_img_tinput_img_
 
double threshold_nms_
 
double threshold_score_
 
bool is_nhwc_
 
std::forward_list< ma_bbox_tresults_
 
- Protected Attributes inherited from ma::Model
Enginep_engine_
 
const char * p_name_
 

Constructor & Destructor Documentation

◆ Detector()

ma::model::Detector::Detector ( Engine engine,
const char *  name,
ma_model_type_t  type 
)

◆ ~Detector()

ma::model::Detector::~Detector ( )
virtual

Member Function Documentation

◆ getConfig()

ma_err_t ma::model::Detector::getConfig ( ma_model_cfg_opt_t  opt,
  ... 
)
overridevirtual

Implements ma::Model.

◆ getInput()

const void * ma::model::Detector::getInput ( )
overridevirtual

Implements ma::Model.

◆ getResults()

const std::forward_list< ma_bbox_t > & ma::model::Detector::getResults ( )

◆ preprocess()

ma_err_t ma::model::Detector::preprocess ( )
overrideprotectedvirtual

Implements ma::Model.

Here is the call graph for this function:

◆ run()

ma_err_t ma::model::Detector::run ( const ma_img_t img)
Here is the call graph for this function:

◆ setConfig()

ma_err_t ma::model::Detector::setConfig ( ma_model_cfg_opt_t  opt,
  ... 
)
overridevirtual

Implements ma::Model.

Member Data Documentation

◆ img_

ma_img_t ma::model::Detector::img_
protected

◆ input_

ma_tensor_t ma::model::Detector::input_
protected

◆ input_img_

const ma_img_t* ma::model::Detector::input_img_
protected

◆ is_nhwc_

bool ma::model::Detector::is_nhwc_
protected

◆ results_

std::forward_list<ma_bbox_t> ma::model::Detector::results_
protected

◆ threshold_nms_

double ma::model::Detector::threshold_nms_
protected

◆ threshold_score_

double ma::model::Detector::threshold_score_
protected