YOLOv10 测试模型性能报错AttributeError: ‘dict‘ object has no attribute ‘shape‘

最新推荐文章于 2025-03-29 12:13:45 发布

每天都在深度学习

最新推荐文章于 2025-03-29 12:13:45 发布

阅读量2.9k

点赞数 14

文章标签： YOLO 前端 javascript

本文链接：https://blog.csdn.net/weixin_58875151/article/details/140259637

版权

当我们在运行YOLOv10的val.py时，发现模型报错AttributeError: 'dict' object has no attribute 'shape'。通过在网上查找相关解决办法，发现原来是没有使用YOLOv10版本的val方式，而是使用了默认v8的方式（因为v10是在v8基础上修改的）。

因此，我们修改代码如下：

rom ultralytics import YOLOv10

model = YOLOv10(model="best.pt")

if __name__ == '__main__':

    model.val(batch=8,
              imgsz=640,
              data='datasets/uw/uw.yaml')

至此，问题解决。

参考博客：yolov10的resume报错解决_yolov10 attributeerror: 'dict' object has no attri-CSDN博客