Video Analysis底层库拆分,sdk的go封装
zhangmeng
2019-11-19 5fe6018c5ce963778de2d31666e61649dd02901c
csrc/buz/yolo/detector.h
New file
@@ -0,0 +1,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