From 33cdaa5782efb2bc2b5ede945771ba647882830d Mon Sep 17 00:00:00 2001 From: Kermit <33612777+keeper-jie@users.noreply.github.com> Date: Sat, 9 Nov 2024 00:55:44 +0800 Subject: [PATCH] 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> --- ultralytics/engine/results.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/ultralytics/engine/results.py b/ultralytics/engine/results.py index 7d8192d6..029e4471 100644 --- a/ultralytics/engine/results.py +++ b/ultralytics/engine/results.py @@ -535,9 +535,9 @@ class Results(SimpleClass): # Plot Detect results if pred_boxes is not None and show_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] - 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() annotator.box_label( box,