Add callback from multitracker to engine

[trackerpp.git] / include / Tracker.h

diff --git a/include/Tracker.h b/include/Tracker.h
index 9eb3139..d528670 100644 (file)
--- a/include/Tracker.h
+++ b/include/Tracker.h
@@ -7,6 +7,7 @@
 #include "Metrics.h"
 #include "SharedPtr.h"
 #include "MultiTracker.h"
+#include "Detector.h"
 
 namespace suanzi {
 
@@ -23,12 +24,14 @@ namespace suanzi {
     class Tracker
     {
     public:
-        Tracker(const cv::Mat& image, int id = 0);
+        Tracker(const cv::Mat& image, const Detection& d, int id = 0);
         virtual ~Tracker();
         void updateState(const cv::Mat& image);
         void addPatch(PatchPtr p);
+        void correct(const cv::Mat& image, const Detection& d);
         TrackerStatus status;
         std::vector<PatchPtr> patches;
+        Detection detection;
 
     private:
         TrackerStatus preStatus;