From b63c8219736016e3b3952465f41abede37a38fbd Mon Sep 17 00:00:00 2001
From: zhangmeng <775834166@qq.com>
Date: 星期三, 11 十二月 2019 11:44:06 +0800
Subject: [PATCH] update
---
csrc/yolo.h | 29 ++++-------------------------
1 files changed, 4 insertions(+), 25 deletions(-)
diff --git a/csrc/yolo.h b/csrc/yolo.h
index caf4e2b..6dc1a0a 100644
--- a/csrc/yolo.h
+++ b/csrc/yolo.h
@@ -1,9 +1,6 @@
#ifndef _cpp_yolo_h_
#define _cpp_yolo_h_
-#include <vector>
-#include <string>
-
#include "struct.h"
struct image_t;
@@ -12,30 +9,12 @@
namespace cppyolo
{
- class sdkyolo{
+ void *init(const char *cfg, const char *weights, const char *name, const int gpu);
+ void release(void *handle);
- public:
- sdkyolo(const char *cfg, const char *weights, const char *name, const int gpu);
- ~sdkyolo();
+ int detect(void *handle, const cIMAGE *img, const float thrsh, const bool use_mean, void **objs, int *objCount);
+ const char* obj_name_by_type(const int typ);
- public:
- int detect(const cIMAGE *img, const float thrsh, const bool use_mean, void **objs, int *objCount);
- const char* obj_name_by_type(const int typ)const;
- private:
- bool init(const char *cfg, const char *weights, const char *name, const int gpu);
- int buffer_to_image(const unsigned char *data, const int w, const int h, const int color_channel);
-
- private:
- Detector *det_;
- cObjInfo *infos_;
- int obj_cnt_;
- std::vector<std::string> names_;
-
- image_t *image_;
- int width_;
- int height_;
- int channel_;
- };
} // namespace cppyolo
--
Gitblit v1.8.0