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

Fix det.summary() bug for ultralytics>=8.2.10 (#13024)

Browse source 
Co-authored-by: UltralyticsAssistant <web@ultralytics.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Co-authored-by: hdnh2006 <henry.matucv@gmail.com>
This commit is contained in:
Henry 2024-05-22 21:46:18 +02:00 • committed by GitHub
parent f0b7c510db
commit d5aff5cc0c
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
ultralytics/engine/results.py
View file

@ -402,8 +402,8 @@ class Results(SimpleClass):
) )
return results return results
data = self.boxes or self.obb
is_obb = self.obb is not None is_obb = self.obb is not None
data = self.obb if is_obb else self.boxes
h, w = self.orig_shape if normalize else (1, 1) h, w = self.orig_shape if normalize else (1, 1)
for i, row in enumerate(data): # xyxy, track_id if tracking, conf, class_id for i, row in enumerate(data): # xyxy, track_id if tracking, conf, class_id
class_id, conf = int(row.cls), round(row.conf.item(), decimals) class_id, conf = int(row.cls), round(row.conf.item(), decimals)