模型训练

  1. 免费数据集网站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,只需要改个几个数字就好啦!其它基本不用管。

参考链接:

  1. link
  2. link
上一篇:Vue入门-指令学习-v-show和v-if


下一篇:MATLAB plot画线的颜色 形状