yolov10的resume报错解决

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

使用yolov10 train跑出来的模型出现这个报错，那么大概率问题出在没有用v10原本的train而是使用默认的v8所以会报错

AttributeError: 'str' object has no attribute 'view'

使用resume报错同上

同一的解决办法是不使用命令行，在vscode中导入环境，使用如下代码

若使用train的模型检测报错 则使用

from ultralytics import YOLOv10
model = YOLOv10("./best.pt")

if __name__ == '__main__':
        model.predict(save=True,source=0) # 0代表打开摄像头，读者可自行更换

若使用resume报错 则使用

from ultralytics import YOLOv10

model = YOLOv10(model="./runs/detect/train5/weights/last.pt")

if __name__ == '__main__':

    model.train(epochs=500,batch=4,imgsz=640,resume=True,data='./data.yaml')

如果有帮助的话记得点一个免费的赞 创作不易，感谢支持🙇‍