X-Git-Url: http://47.100.26.94:8080/?a=blobdiff_plain;f=include%2FTracker.h;h=d5286706d010f64a25005a518e8f2a8b8010e99f;hb=a80807eb35cc0ff70a9afdd6b7db2f38cb686683;hp=9eb3139c2b59132173f408424113dfbe43cdfe22;hpb=3aa517d206c44156fe86697aeadc5f75ea212329;p=trackerpp.git

diff --git a/include/Tracker.h b/include/Tracker.h
index 9eb3139..d528670 100644
--- 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;