YOLOv10训练报错

YOLOv10是在YOLOv8基础上修改的，即：训练方法和过程是相同的。

但按照v8训练程序train.py，如下所示，直接训练：

from ultralytics import YOLO# Load a model
model = YOLO("ultralytics/cfg/models/v8/yolov8n.yaml")  # build a new 
# train
model.train(data='dataset/data.yaml',cache=False,imgsz=640,epochs=100,batch=16,close_mosaic=0,workers=4,device='0',optimizer='SGD', # using SGDamp=False, # close ampproject='runs/train',name='exp',)

会提示以下错误：

AttributeError: ‘str’ object has no attribute ‘view’

解决方法

把代码中的YOLO改为YOLOv10即可，注意下面代码第一行和第三行。

from ultralytics import YOLOv10   # 修改处 1# Load a model
model = YOLOv10("ultralytics/cfg/models/v8/yolov8n.yaml") # 修改处 2
# train
model.train(data='dataset/data.yaml',cache=False,imgsz=640,epochs=100,batch=16,close_mosaic=0,workers=4,device='0',optimizer='SGD', # using SGDamp=False, # close ampproject='runs/train',name='exp',)