rachana219 commited on
Commit
9890183
·
1 Parent(s): 941336a

Update track.py

Browse files
Files changed (1) hide show
  1. track.py +5 -2
track.py CHANGED
@@ -182,14 +182,17 @@ def run(
182
  p = Path(p) # to Path
183
  s += f'{i}: '
184
  txt_file_name = p.name
185
- save_path = str(save_dir / p.name) # im.jpg, vid.mp4, ...
 
 
186
  else:
187
  p, im0, _ = path, im0s.copy(), getattr(dataset, 'frame', 0)
188
  p = Path(p) # to Path
189
  # video file
190
  if source.endswith(VID_FORMATS):
191
  txt_file_name = p.stem
192
- save_path = str(save_dir / p.name) # im.jpg, vid.mp4, ...
 
193
  # folder with imgs
194
  else:
195
  txt_file_name = p.parent.name # get folder name containing current img
 
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:
189
  p, im0, _ = path, im0s.copy(), getattr(dataset, 'frame', 0)
190
  p = Path(p) # to Path
191
  # video file
192
  if source.endswith(VID_FORMATS):
193
  txt_file_name = p.stem
194
+ save_path = str(save_dir / filename) # im.jpg, vid.mp4, ...
195
+ LOGGER.info(f"p.name is {p.name}, save_path value is {save_path}")
196
  # folder with imgs
197
  else:
198
  txt_file_name = p.parent.name # get folder name containing current img