rachana219 commited on
Commit
3a78219
·
1 Parent(s): 9890183

Update track.py

Browse files
Files changed (1) hide show
  1. track.py +1 -1
track.py CHANGED
@@ -175,6 +175,7 @@ def run(
175
  p = non_max_suppression(preds, conf_thres, iou_thres, classes, agnostic_nms, max_det=max_det)
176
 
177
  # Process detections
 
178
  for i, det in enumerate(p): # detections per image
179
  seen += 1
180
  if webcam: # bs >= 1
@@ -182,7 +183,6 @@ def run(
182
  p = Path(p) # to Path
183
  s += f'{i}: '
184
  txt_file_name = p.name
185
- filename = 'out.mp4'
186
  save_path = str(save_dir / filename) # im.jpg, vid.mp4, ...
187
 
188
  else:
 
175
  p = non_max_suppression(preds, conf_thres, iou_thres, classes, agnostic_nms, max_det=max_det)
176
 
177
  # Process detections
178
+ filename = 'out.mp4'
179
  for i, det in enumerate(p): # detections per image
180
  seen += 1
181
  if webcam: # bs >= 1
 
183
  p = Path(p) # to Path
184
  s += f'{i}: '
185
  txt_file_name = p.name
 
186
  save_path = str(save_dir / filename) # im.jpg, vid.mp4, ...
187
 
188
  else: