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: