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

#include <ma_model_pose_detector.h>

Inheritance diagram for ma::model::PoseDetector:
Collaboration diagram for ma::model::PoseDetector:

Public Member Functions

 PoseDetector (Engine *engine, const char *name, ma_model_type_t type)
 
virtual ~PoseDetector ()
 
const std::forward_list< ma_keypoint3f_t > & getResults () const
 
ma_err_t run (const ma_img_t *img)
 
const void * getInput () override
 
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_
 
float threshold_nms_
 
float threshold_score_
 
bool is_nhwc_
 
std::forward_list< ma_keypoint3f_t > results_
 
- Protected Attributes inherited from ma::Model
Enginep_engine_
 
const char * p_name_
 

Constructor & Destructor Documentation

◆ PoseDetector()

ma::model::PoseDetector::PoseDetector ( Engine engine,
const char *  name,
ma_model_type_t  type 
)
Here is the call graph for this function:

◆ ~PoseDetector()

ma::model::PoseDetector::~PoseDetector ( )
virtual

Member Function Documentation

◆ getConfig()

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

Implements ma::Model.

◆ getInput()

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

Implements ma::Model.

◆ getResults()

const std::forward_list< ma_keypoint3f_t > & ma::model::PoseDetector::getResults ( ) const

◆ preprocess()

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

Implements ma::Model.

Here is the call graph for this function:

◆ run()

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

◆ setConfig()

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

Implements ma::Model.

Member Data Documentation

◆ img_

ma_img_t ma::model::PoseDetector::img_
protected

◆ input_

ma_tensor_t ma::model::PoseDetector::input_
protected

◆ input_img_

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

◆ is_nhwc_

bool ma::model::PoseDetector::is_nhwc_
protected

◆ results_

std::forward_list<ma_keypoint3f_t> ma::model::PoseDetector::results_
protected

◆ threshold_nms_

float ma::model::PoseDetector::threshold_nms_
protected

◆ threshold_score_

float ma::model::PoseDetector::threshold_score_
protected