Initialize angles vector dynamically

diff --git a/src/kcf.cpp b/src/kcf.cpp
index 2a778227c27ce1539a721989e581f4aebb551961..b6ead00298fdd371d8e40b6ba841d3507a9f0f27 100644 (file)
--- a/src/kcf.cpp
+++ b/src/kcf.cpp
@@ -180,6 +180,10 @@ void KCF_Tracker::init(cv::Mat &img, const cv::Rect &bbox, int fit_size_x, int f
     for (int i = -int(p_num_scales) / 2; i <= int(p_num_scales) / 2; ++i)
         p_scales.push_back(std::pow(p_scale_step, i));
 
+    p_angles.clear();
+    for (int i = -int(p_num_angles) / 2; i <= int(p_num_angles) / 2; ++i)
+        p_angles.push_back(i * p_angle_step);
+
 #ifdef CUFFT
     if (m_use_linearkernel) {
         std::cerr << "cuFFT supports only Gaussian kernel." << std::endl;

diff --git a/src/kcf.h b/src/kcf.h
index d0520129d1350f44b1f57cbabcf24b17699939f0..70662d7920ea7849038ea5bfef9aa5a91227c8d3 100644 (file)
--- a/src/kcf.h
+++ b/src/kcf.h
@@ -110,7 +110,7 @@ private:
 
     const uint p_num_angles = 1;
     const int p_angle_step = 10;
-    std::vector<double> p_angles = {0};
+    std::vector<double> p_angles;
 
     const int p_num_of_feats = 31 + (m_use_color ? 3 : 0) + (m_use_cnfeat ? 10 : 0);
     cv::Size feature_size;