New file |
| | |
| | | #ifndef _c_wrapper_yolo_detector_h_ |
| | | #define _c_wrapper_yolo_detector_h_ |
| | | |
| | | #include "../base.hpp" |
| | | |
| | | struct _cObjInfo; |
| | | struct _cIMAGE; |
| | | |
| | | struct image_t; |
| | | |
| | | |
| | | namespace csdk_wrap{ |
| | | void* init_yolo_detector(const char *cfg, const char *weights, const char *name, |
| | | const int gpu_index, VecFunc &vec); |
| | | image_t* buffer_to_image(const unsigned char *data, const int w, const int h, const int color_channel); |
| | | cObjInfo* yolo_detect(void *handle,int *objCount, const cIMAGE *img, const float thrsh, const bool use_mean); |
| | | |
| | | const char* yolo_obj_name_by_type(const int typ); |
| | | } |
| | | |
| | | #endif |