From b87bfa32c36f582d21ac3da7b21d9d9178d339ba Mon Sep 17 00:00:00 2001
From: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Thu, 12 Dec 2019 13:56:56 -0800
Subject: [PATCH] updates

---
 detect.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/detect.py b/detect.py
index 2593406c..6da45c3b 100644
--- a/detect.py
+++ b/detect.py
@@ -125,6 +125,8 @@ def detect(save_txt=False, save_img=False):
             # Stream results
             if view_img:
                 cv2.imshow(p, im0)
+                if cv2.waitKey(1) == ord('q'):  # q to quit
+                    raise StopIteration
 
             # Save results (image with detections)
             if save_img: