updates

2019-12-12 13:56:56 -08:00 · 2019-12-12 13:56:56 -08:00 · b87bfa32c3
commit b87bfa32c3
parent 8d8daff390
1 changed files with 2 additions and 0 deletions
--- 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: