From 15bdea757aa77aada1412724c6d383c1bfb5ba11 Mon Sep 17 00:00:00 2001 From: kangsanha <68986305+sanha9999@users.noreply.github.com> Date: Tue, 4 Jun 2024 20:18:04 +0900 Subject: [PATCH] fix onnx model val.py AssertionError (#190) --- ultralytics/models/yolov10/val.py | 1 + 1 file changed, 1 insertion(+) 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: