Spaces:
Sleeping
Sleeping
Commit
·
609e92c
1
Parent(s):
8a63318
Added box name
Browse files
app.py
CHANGED
@@ -74,7 +74,7 @@ def inference(input_img, conf_thres, iou_thres):
|
|
74 |
# Write results
|
75 |
for *xyxy, conf, cls in reversed(det):
|
76 |
c = int(cls) # integer class
|
77 |
-
label = '{names[c]} {conf:.2f}'
|
78 |
annotator.box_label(xyxy, label, color=colors(c, True))
|
79 |
|
80 |
return im0
|
|
|
74 |
# Write results
|
75 |
for *xyxy, conf, cls in reversed(det):
|
76 |
c = int(cls) # integer class
|
77 |
+
label = f'{names[c]} {conf:.2f}'
|
78 |
annotator.box_label(xyxy, label, color=colors(c, True))
|
79 |
|
80 |
return im0
|