upgrade opencv to 3.3.1, complete patch similarity

[trackerpp.git] / include / PredictorWrapper.h

diff --git a/include/PredictorWrapper.h b/include/PredictorWrapper.h
index fab9497..41f818f 100644 (file)
--- a/include/PredictorWrapper.h
+++ b/include/PredictorWrapper.h
@@ -3,6 +3,7 @@
 
 #include "SharedPtr.h"
 #include <boost/python.hpp>
+#include <vector>
 
 namespace suanzi {
 
@@ -15,8 +16,8 @@ namespace suanzi {
     public:
         static PredictorWrapperPtr create(const std::string& fname);
         ~PredictorWrapper(){}
-        void dump() { this->dump_func(); }
-        void predict() { this->predict_func();}
+        void dump();
+        double predict(int index, const std::vector<double>& f);
 
     private:
         PredictorWrapper(const std::string& fname);