nsaunier/traffic-intelligence

bug at the end of video

Commit 27b206d118b7 · Nicolas Saunier · 2024-05-31 16:52 -0400

Changeset

27b206d118b73c7442d78d6bc098b831a5999cb3

View source at this commit

Comments

No comments yet.

Log in to comment

Diff

diff --git a/scripts/dltrack.py b/scripts/dltrack.py
--- a/scripts/dltrack.py
+++ b/scripts/dltrack.py
@@ -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()