FFT does not take whole Scale_vars struct

[hercules2020/kcf.git] / src / kcf.h

diff --git a/src/kcf.h b/src/kcf.h
index 1c0f5d22295990ebc43519c7836b9a79c926886c..8f217f473a9cd8ece6d9748ef3e80e3b38edde11 100644 (file)
--- a/src/kcf.h
+++ b/src/kcf.h
@@ -69,6 +69,11 @@ public:
 #else
     bool m_use_big_batch {false};
 #endif
+#ifdef CUFFT
+    bool m_use_cuda {true};
+#else
+    bool m_use_cuda {false};
+#endif
 
     /*
     padding             ... extra area surrounding the target           (1.5)
@@ -123,7 +128,7 @@ private:
     int p_num_of_feats;
     int p_roi_height, p_roi_width;
 
-    std::vector<Scale_vars> scale_vars;
+    std::vector<Scale_vars> p_scale_vars;
 
     //model
     ComplexMat p_yf;
@@ -134,7 +139,7 @@ private:
     //helping functions
     void scale_track(Scale_vars & vars, cv::Mat & input_rgb, cv::Mat & input_gray, double scale);
     cv::Mat get_subwindow(const cv::Mat & input, int cx, int cy, int size_x, int size_y);
-    void gaussian_shaped_labels(Scale_vars & vars, double sigma, int dim1, int dim2);
+    cv::Mat gaussian_shaped_labels(double sigma, int dim1, int dim2);
     void gaussian_correlation(struct Scale_vars &vars, const ComplexMat & xf, const ComplexMat & yf, double sigma, bool auto_correlation = false);
     cv::Mat circshift(const cv::Mat & patch, int x_rot, int y_rot);
     cv::Mat cosine_window_function(int dim1, int dim2);