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',)