Add Sconscript for unit test
[trackerpp.git] / include / Detector.h
diff --git a/include/Detector.h b/include/Detector.h
new file mode 100644 (file)
index 0000000..edf875c
--- /dev/null
@@ -0,0 +1,41 @@
+#ifndef _DETECTOR_H_
+#define _DETECTOR_H_
+
+#include "VideoReader.h"
+#include "SharedPtr.h"
+
+namespace suanzi {
+
+    struct Detection;
+
+    TK_DECLARE_PTR(Detector);
+    class Detector
+    {
+    public:
+        Detector();
+        virtual ~Detector();
+        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;
+    };
+
+}
+
+
+#endif /* _DETECTOR_H_ */
+
+