Spaces:
Running
on
Zero
Running
on
Zero
Fixing tracking
Browse files
app.py
CHANGED
@@ -243,11 +243,12 @@ def get_tracker(tracker: str, fps: float):
|
|
243 |
|
244 |
|
245 |
def update_tracker(tracker, detections, frame):
|
246 |
-
|
|
|
247 |
return tracker.update(detections)
|
248 |
-
elif
|
249 |
return tracker.update(detections, frame)
|
250 |
-
elif
|
251 |
return tracker.update_with_detections(detections)
|
252 |
else:
|
253 |
raise ValueError(f"Invalid tracker: {tracker}")
|
|
|
243 |
|
244 |
|
245 |
def update_tracker(tracker, detections, frame):
|
246 |
+
tracker_name = tracker.__class__.__name__
|
247 |
+
if tracker_name == "SORTTracker":
|
248 |
return tracker.update(detections)
|
249 |
+
elif tracker_name == "DeepSORTTracker":
|
250 |
return tracker.update(detections, frame)
|
251 |
+
elif tracker_name == "ByteTrack":
|
252 |
return tracker.update_with_detections(detections)
|
253 |
else:
|
254 |
raise ValueError(f"Invalid tracker: {tracker}")
|