| | |
| | | |
| | | #include "csrc/yolo.cpp" |
| | | |
| | | using namespace cppyolo; |
| | | |
| | | void* init(const char *cfg, const char *weights, const char *name, const int gpu){ |
| | | return new sdkyolo(cfg, weights, name, gpu); |
| | | return cppyolo::init(cfg, weights, name, gpu); |
| | | } |
| | | |
| | | void release(void *handle){ |
| | | if (handle){ |
| | | sdkyolo *h = (sdkyolo*)handle; |
| | | delete h; |
| | | cppyolo::release(handle); |
| | | } |
| | | } |
| | | |
| | |
| | | const float thrsh, const int use_mean, |
| | | void **objInfos, int *objCount){ |
| | | |
| | | sdkyolo *y = (sdkyolo*)handle; |
| | | cIMAGE img{(unsigned char*)data, w, h, c}; |
| | | return y->detect(&img, thrsh, use_mean, objInfos, objCount); |
| | | return cppyolo::detect(handle, &img, thrsh, use_mean, objInfos, objCount); |
| | | } |
| | | |
| | | const char* obj_name_by_type(void *handle, const int typ){ |
| | | sdkyolo *h = (sdkyolo*)handle; |
| | | return h->obj_name_by_type(typ); |
| | | const char* obj_name_by_type(const int typ){ |
| | | |
| | | return cppyolo::obj_name_by_type(typ); |
| | | } |
| | | |