Fix `conf` overwrite in results.py (#17384)

Co-authored-by: Ultralytics Assistant <135830346+UltralyticsAssistant@users.noreply.github.com>
Co-authored-by: Burhan <62214284+Burhan-Q@users.noreply.github.com>
pull/17416/head^2
Kermit 3 weeks ago committed by GitHub
parent cc81b06679
commit 33cdaa5782
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
  1. 4
      ultralytics/engine/results.py

@ -535,9 +535,9 @@ class Results(SimpleClass):
# Plot Detect results # Plot Detect results
if pred_boxes is not None and show_boxes: if pred_boxes is not None and show_boxes:
for i, d in enumerate(reversed(pred_boxes)): for i, d in enumerate(reversed(pred_boxes)):
c, conf, id = int(d.cls), float(d.conf) if conf else None, None if d.id is None else int(d.id.item()) c, d_conf, id = int(d.cls), float(d.conf) if conf else None, None if d.id is None else int(d.id.item())
name = ("" if id is None else f"id:{id} ") + names[c] name = ("" if id is None else f"id:{id} ") + names[c]
label = (f"{name} {conf:.2f}" if conf else name) if labels else None label = (f"{name} {d_conf:.2f}" if conf else name) if labels else None
box = d.xyxyxyxy.reshape(-1, 4, 2).squeeze() if is_obb else d.xyxy.squeeze() box = d.xyxyxyxy.reshape(-1, 4, 2).squeeze() if is_obb else d.xyxy.squeeze()
annotator.box_label( annotator.box_label(
box, box,

Loading…
Cancel
Save