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

Update detect_deepsort.py

Browse files
Files changed (1) hide show
  1. detect_deepsort.py +2 -2
detect_deepsort.py CHANGED
@@ -110,8 +110,8 @@ def draw_boxes(frame, bbox_xyxy, draw_trails, identities=None, categories=None,
110
  cv2.line(frame, data_deque[id][i - 1], data_deque[id][i], color, thickness)
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)
 
110
  cv2.line(frame, data_deque[id][i - 1], data_deque[id][i], color, thickness)
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)