SSCMA-Micro CPP SDK
v2.0.0
SSCMA-Micro is a cross-platform machine learning inference framework designed for embedded devices.
|
#include <ma_model_yolov5.h>
Public Member Functions | |
YoloV5 (Engine *engine) | |
~YoloV5 () | |
Public Member Functions inherited from ma::model::Detector | |
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) |
Static Public Member Functions | |
static bool | isValid (Engine *engine) |
Protected Member Functions | |
ma_err_t | postprocess () override |
ma_err_t | generalPostProcess () |
ma_err_t | nmsPostProcess () |
Protected Member Functions inherited from ma::model::Detector | |
ma_err_t | preprocess () override |
Protected Member Functions inherited from ma::Model | |
ma_err_t | underlyingRun () |
Additional Inherited Members | |
Protected Attributes inherited from ma::model::Detector | |
ma_tensor_t | input_ |
ma_img_t | img_ |
const ma_img_t * | input_img_ |
double | threshold_nms_ |
double | threshold_score_ |
bool | is_nhwc_ |
std::forward_list< ma_bbox_t > | results_ |
Protected Attributes inherited from ma::Model | |
Engine * | p_engine_ |
const char * | p_name_ |
ma::model::YoloV5::YoloV5 | ( | Engine * | engine | ) |
ma::model::YoloV5::~YoloV5 | ( | ) |
|
protected |
|
static |
|
protected |