ntsc207 commited on
Commit
94e9b34
·
verified ·
1 Parent(s): 10d24ec

Update detect_deepsort.py

Browse files
Files changed (1) hide show
  1. detect_deepsort.py +1 -1
detect_deepsort.py CHANGED
@@ -111,7 +111,7 @@ def draw_boxes(frame, bbox_xyxy, draw_trails, identities=None, categories=None,
111
  return frame
112
 
113
  #@spaces.GPU(duration=120)
114
- #@smart_inference_mode()
115
  def run_deepsort(
116
  weights=ROOT / 'yolo.pt', # model path or triton URL
117
  source=ROOT / 'data/images', # file/dir/URL/glob/screen/0(webcam)
 
111
  return frame
112
 
113
  #@spaces.GPU(duration=120)
114
+ @smart_inference_mode()
115
  def run_deepsort(
116
  weights=ROOT / 'yolo.pt', # model path or triton URL
117
  source=ROOT / 'data/images', # file/dir/URL/glob/screen/0(webcam)