HongChuang/ultralytics-ascend
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 3

Fix bug in Results.summary() for masks and keypoints (#12629)

Browse source 
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
This commit is contained in:
Max 2024-05-12 20:13:49 +03:00 • committed by GitHub
parent 6909f08dc5
commit cf24349efb
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 3 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

6
ultralytics/engine/results.py
View file

@ -407,9 +407,9 @@ class Results(SimpleClass):
class_id, conf = int(row.cls), round(row.conf.item(), decimals) class_id, conf = int(row.cls), round(row.conf.item(), decimals)
box = (row.xyxyxyxy if is_obb else row.xyxy).squeeze().reshape(-1, 2).tolist() box = (row.xyxyxyxy if is_obb else row.xyxy).squeeze().reshape(-1, 2).tolist()
xy = {} xy = {}
for i, b in enumerate(box): for j, b in enumerate(box):
xy[f"x{i + 1}"] = round(b[0] / w, decimals) xy[f"x{j + 1}"] = round(b[0] / w, decimals)
xy[f"y{i + 1}"] = round(b[1] / h, decimals) xy[f"y{j + 1}"] = round(b[1] / h, decimals)
result = {"name": self.names[class_id], "class": class_id, "confidence": conf, "box": xy} result = {"name": self.names[class_id], "class": class_id, "confidence": conf, "box": xy}
if data.is_track: if data.is_track:
result["track_id"] = int(row.id.item()) # track ID result["track_id"] = int(row.id.item()) # track ID