updates

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2019-09-02 20:53:49 +02:00
parent 0d5bf11fa5
commit b76962771e
2 changed files with 2 additions and 1 deletions
@@ -146,6 +146,7 @@ class LoadWebcam:  # for inference
    def __next__(self):
        self.count += 1
        if cv2.waitKey(1) == 27:  # esc to quit
+            self.cap.release()
            cv2.destroyAllWindows()
            raise StopIteration