From 5fe6018c5ce963778de2d31666e61649dd02901c Mon Sep 17 00:00:00 2001
From: zhangmeng <775834166@qq.com>
Date: 星期二, 19 十一月 2019 14:56:18 +0800
Subject: [PATCH] update
---
csrc/buz/yolo/detector.h | 21 +++++++++++++++++++++
1 files changed, 21 insertions(+), 0 deletions(-)
diff --git a/csrc/buz/yolo/detector.h b/csrc/buz/yolo/detector.h
new file mode 100644
index 0000000..1f263b8
--- /dev/null
+++ b/csrc/buz/yolo/detector.h
@@ -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
\ No newline at end of file
--
Gitblit v1.8.0