Leave 1px borders in visual debug window

[hercules2020/kcf.git] / main_vot.cpp

diff --git a/main_vot.cpp b/main_vot.cpp
index 32e54542db5bbdf5aa17cf5e30b8a46ed4fb5adc..53d2ca62b55f922757dcde3d95eee5635f465b33 100644 (file)
--- a/main_vot.cpp
+++ b/main_vot.cpp
@@ -189,10 +189,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;