tuansunday05 commited on
Commit
a4db620
·
verified ·
1 Parent(s): 6245b9e

Update detect_strongsort.py

Browse files
Files changed (1) hide show
  1. detect_strongsort.py +1 -1
detect_strongsort.py CHANGED
@@ -264,7 +264,7 @@ def run(
264
 
265
  if save_img or save_crop or view_img: # Add bbox to image
266
  c = int(cls) # integer class
267
- label = None if hide_labels else (f' { id } {names[c]}' if hide_conf else f' { id } {names[c]} {conf:.2f}')
268
  plot_one_box(xyxy, im0, label=label, color=colors[int(cls)], line_thickness=2)
269
  if save_crop:
270
  save_one_box(xyxy, imc, file=save_dir / 'crops' / names[c] / f'{p.stem}.jpg', BGR=True)
 
264
 
265
  if save_img or save_crop or view_img: # Add bbox to image
266
  c = int(cls) # integer class
267
+ label = None if hide_labels else ( str(id) + ' ' + names[c] if hide_conf else f' { id } {names[c]} {conf:.2f}')
268
  plot_one_box(xyxy, im0, label=label, color=colors[int(cls)], line_thickness=2)
269
  if save_crop:
270
  save_one_box(xyxy, imc, file=save_dir / 'crops' / names[c] / f'{p.stem}.jpg', BGR=True)