diff --git a/ultralytics/models/yolov10/val.py b/ultralytics/models/yolov10/val.py index 1814a2b8..19a019c8 100644 --- a/ultralytics/models/yolov10/val.py +++ b/ultralytics/models/yolov10/val.py @@ -14,6 +14,7 @@ class YOLOv10DetectionValidator(DetectionValidator): if isinstance(preds, (list, tuple)): preds = preds[0] + # Acknowledgement: Thanks to sanha9999 in #190 and #181! if preds.shape[-1] == 6: return preds else: