1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
| #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
|
|
