AttributeError: 'dict' object has no attribute 'shape'

THU-MIG / yolov10

YOLOv10: Real-Time End-to-End Object Detection

https://arxiv.org/abs/2405.14458

GNU Affero General Public License v3.0

8k stars 662 forks source link

AttributeError: 'dict' object has no attribute 'shape' #275

Open jwy-john opened 2 weeks ago

jwy-john commented 2 weeks ago

File "/home/hdou/OB/v10-main/ultralytics/utils/ops.py", line 216, in non_max_suppression bs = prediction.shape[0] # batch size AttributeError: 'dict' object has no attribute 'shape' Sentry is attempting to send 2 pending events

在模型val的时候出现了这个问题，这个是什么情况呢

leonnil commented 1 week ago

Thank you for your interest! Does this issue https://github.com/THU-MIG/yolov10/issues/46 help?