- 免费数据集网站Roboflow,—键导出Voc、coco、Yolo、Csv等格式。
随便下载了一个数据集用它导出YOLO的数据集,自动给转换成txt的格式,yaml文件也已经配置好了,直接用就可以。
2. 模型下载,就是在上面的链接,即训练代码链接,下载代码即可
git clone https://github.com/ultralytics/ultralytics.git
3. 训练代码,train.py如下:
import warnings
warnings.filterwarnings('ignore')
from ultralytics import YOLO
if __name__ == '__main__':
model = YOLO(r'xxx\yolo\yolov11\ultralytics-main\datasets\yolo11.yaml') # 地址改成自己的
model.train(data=r'xxx\yolo\yolov11\ultralytics-main\datasets\data.yaml',
cache=False,
imgsz=640,
epochs=100,
single_cls=False, # 是否是单类别检测
batch=8,
close_mosaic=10,
workers=0,
device='0',
optimizer='SGD',
amp=True,
project='runs/train',
name='exp',
)
可以说,只要你会YOLOv8从训练到部署,切换到YOLOv11,只需要改个几个数字就好啦!其它基本不用管。
参考链接:
- link
- link