fix: 🐞 yolo obb hasstr added into condition

Signed-off-by: Onuralp SEZER <thunderbirdtr@gmail.com>
This commit is contained in:
Onuralp SEZER 2024-05-22 19:48:39 +03:00
parent 2bd4348e5d
commit 251185ba67
No known key found for this signature in database
GPG Key ID: CF0835DFDF14CA38
1 changed files with 1 additions and 1 deletions

View File

@ -240,7 +240,7 @@ class Detections:
Class names values can be accessed using `detections["class_name"]`.
""" # noqa: E501 // docs
if "obb" in ultralytics_results and ultralytics_results.obb is not None:
if hasattr(ultralytics_results, "obb") and ultralytics_results.obb is not None:
class_id = ultralytics_results.obb.cls.cpu().numpy().astype(int)
class_names = np.array([ultralytics_results.names[i] for i in class_id])
oriented_box_coordinates = ultralytics_results.obb.xyxyxyxy.cpu().numpy()