From b63c8219736016e3b3952465f41abede37a38fbd Mon Sep 17 00:00:00 2001
From: zhangmeng <775834166@qq.com>
Date: 星期三, 11 十二月 2019 11:44:06 +0800
Subject: [PATCH] update
---
cyolo.cpp | 15 ++++++---------
1 files changed, 6 insertions(+), 9 deletions(-)
diff --git a/cyolo.cpp b/cyolo.cpp
index 4a4124c..805393a 100644
--- a/cyolo.cpp
+++ b/cyolo.cpp
@@ -14,16 +14,14 @@
#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);
}
}
@@ -32,13 +30,12 @@
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);
}
--
Gitblit v1.8.0