Move visual debug after all calculations

[hercules2020/kcf.git] / main_vot.cpp

diff --git a/main_vot.cpp b/main_vot.cpp
index 65b982f4108b8286c4ee7377b7b99d537ffe0c14..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"
@@ -191,8 +190,10 @@ int main(int argc, char *argv[])
                 cv::line(image, vertices[i], vertices[(i + 1) % 4], cv::Scalar(0, 255, 0), 2);
             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;