qubvel-hf HF Staff commited on
Commit
a7f1cbc
·
1 Parent(s): 72f165b

Fixing tracking

Browse files
Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -243,11 +243,12 @@ def get_tracker(tracker: str, fps: float):
243
 
244
 
245
  def update_tracker(tracker, detections, frame):
246
- if isinstance(tracker, trackers.SORTTracker):
 
247
  return tracker.update(detections)
248
- elif isinstance(tracker, trackers.DeepSORTTracker):
249
  return tracker.update(detections, frame)
250
- elif isinstance(tracker, sv.ByteTrack):
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}")