yolov10 训练

1、GitHub 上拉代码

地址：https://github.com/THU-MIG/yolov10/tree/main

Ultralytics 文档：https://docs.ultralytics.com/models/yolov10/

2、如果你已经配置好了Ultralytics 环境

那么直接新建一个train.py文件

from ultralytics import YOLOv10
import torch
import os

# 加载模型
# model = YOLO("yolov10s.pt")  # 从头开始构建新模型
model = YOLOv10('ultralytics/cfg/models/v10/yolov10s.yaml')
if __name__ == '__main__':
    # Use the model
    # results = model.train(data="datasets/person.yaml",model=r"D:\yolov8\ultralytics\ultralytics\cfg\models\v8\myyolov8.yaml", epochs=100, batch=32,device=0)             # 训练模型
    results = model.train(data="datasets/coco.yaml", epochs=300,
                          batch=32, device=0)  # 训练模型

这个数据加载方式和之前Ultralytics 中v8的配置是一样的

3、YoloV10训练出现AttributeError: ‘str‘ object has no attribute ‘view

就按照上面的代码来运行，这个问题是你用老的代码去运行就会这样

4、总结

直接换掉模型结构就可以很快从yolov8迁移过来，用这个Ultralytics 工具