diff scripts/dltrack.py @ 1268:27b206d118b7

bug at the end of video
author Nicolas Saunier <nicolas.saunier@polymtl.ca>
date Fri, 31 May 2024 16:52:51 -0400
parents 0f5bebd62a55
children 20a5e1292321
line wrap: on
line diff
--- a/scripts/dltrack.py	Wed May 29 09:52:42 2024 -0400
+++ b/scripts/dltrack.py	Fri May 31 16:52:51 2024 -0400
@@ -134,7 +134,8 @@
             break
     frameNum += 1
     success, frame = capture.read()
-    results = model.track(source=frame, persist=True)
+    if success:
+        results = model.track(source=frame, persist=True)
 capture.release()
 cv2.destroyAllWindows()