xclele/yolov10
1
0
Fork 0
mirror of https://github.com/THU-MIG/yolov10.git synced 2025-05-23 05:24:22 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix onnx model val.py AssertionError (#190)

Browse Source
This commit is contained in:
kangsanha 2024-06-04 20:18:04 +09:00 committed by GitHub
parent ba0e6cbc1a
commit 15bdea757a
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
ultralytics/models/yolov10/val.py
View File

@ -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: