unify the interface with detector

[trackerpp.git] / src / Detector.h

diff --git a/src/Detector.h b/src/Detector.h
index 5cf08c1..edf875c 100644 (file)
--- a/src/Detector.h
+++ b/src/Detector.h
@@ -6,13 +6,31 @@
 
 namespace suanzi {
 
+    struct Detection;
+
     TK_DECLARE_PTR(Detector);
     class Detector
     {
     public:
         Detector();
         virtual ~Detector();
-        void detect(cv::Mat& frame){};
+        unsigned int detect(cv::Mat& frame, Detection* detections){return 1;}
+    };
+
+    struct Detection
+    {
+        // 检测目标的类型，目前只有一个类human，默认等于0。为以后可以检测的更多类别预留
+        unsigned int object_type;
+        // 检测目标的分数，可以不填
+        float score;
+        // 检测目标的坐标，包括物体中心的x、y坐标，物体的高和宽
+        unsigned int center_x;
+        unsigned int center_y;
+        unsigned int height;
+        unsigned int width;
+        // 检测目标的特征向量
+        unsigned int feature_size;
+        float * feature;
     };
 
 }