【YOLOv10训练】：报错AttributeError: ‘str‘ object has no attribute ‘view‘ 解决方法

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 SGD
                amp=False, # close amp
                project='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/v10/yolov10n.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 SGD
                amp=False, # close amp
                project='runs/train',
                name='exp',
                )