#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