Refactor point wrapping in frequency domain to a function

[hercules2020/kcf.git] / main_vot.cpp

diff --git a/main_vot.cpp b/main_vot.cpp
index 32e54542db5bbdf5aa17cf5e30b8a46ed4fb5adc..67b180d740d247ef351659651221f751f5324c35 100644 (file)
--- a/main_vot.cpp
+++ b/main_vot.cpp
@@ -61,7 +61,6 @@ int main(int argc, char *argv[])
             break;
         case 'p':
             tracker.m_visual_debug = true;
-            visualize_delay = 500;
             break;
         case 'h':
             std::cerr << "Usage: \n"
@@ -189,10 +188,12 @@ int main(int argc, char *argv[])
 
             for (int i = 0; i < 4; i++)
                 cv::line(image, vertices[i], vertices[(i + 1) % 4], cv::Scalar(0, 255, 0), 2);
-            cv::imshow("output", image);
+            cv::imshow("KCF output", image);
             int ret = cv::waitKey(visualize_delay);
-            if (visualize_delay > 0 && ret != -1 && ret < 128)
+            if ((visualize_delay > 0 && ret != -1 && ret < 128) ||
+                (visualize_delay == 0 && (ret == 27 /*esc*/ || ret == 'q')))
                 break;
+
         }
 
 //        std::stringstream s;