changeset 1268:27b206d118b7 default tip

bug at the end of video
author Nicolas Saunier <nicolas.saunier@polymtl.ca>
date Fri, 31 May 2024 16:52:51 -0400
parents ad60e5adf084
children
files scripts/dltrack.py
diffstat 1 files changed, 2 insertions(+), 1 deletions(-) [+]
line wrap: on
line diff
diff -r ad60e5adf084 -r 27b206d118b7 scripts/dltrack.py
--- 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()