Spaces:
Running
on
Zero
Running
on
Zero
Update detect_strongsort.py
Browse files- 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 (
|
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)
|