27 lines
1.2 KiB
C
27 lines
1.2 KiB
C
#ifndef __INCLUDE_NN_DETECT_API_H_
|
|
#define __INCLUDE_NN_DETECT_API_H_
|
|
|
|
#ifdef __cplusplus
|
|
extern "C"{
|
|
#endif
|
|
|
|
#include "nn_detect_common.h"
|
|
|
|
det_status_t det_set_model(det_model_type modelType); ///<set network type,this will trigger init process
|
|
det_status_t det_get_model_size(det_model_type modelType, int *width, int *height, int *channel);
|
|
det_status_t det_set_param(det_model_type modelType, det_param_t param);
|
|
det_status_t det_get_param(det_model_type modelType, det_param_t *param);
|
|
det_status_t det_set_input(input_image_t imageData, det_model_type modelType); ///<set input image to network
|
|
det_status_t det_get_result(pDetResult resultData, det_model_type modelType); ///<get detection result,this will block until process end
|
|
det_status_t det_release_model(det_model_type modelType); ///<get detection result,this will block until process end
|
|
|
|
// for async
|
|
det_status_t det_trigger_inference(input_image_t imageData, det_model_type modelType);
|
|
det_status_t det_get_inference_result(pDetResult resultData, det_model_type modelType);
|
|
//det_status_t det_set_data_to_NPU(input_image_t imageData, det_model_type modelType);
|
|
|
|
|
|
#ifdef __cplusplus
|
|
}
|
|
#endif
|
|
#endif |