YOLOv5训练自己的数据集

YOLOv5训练自己的数据集

1.使用labelimg对图片进行标注。

此处只提供安装方法，使用方法后期更新或自行查找。
使用清华镜像源进行安装，python版本要低于3.10，否则会闪退。
虚拟环境或者终端运行：

pip install -i https://pypi.tuna.tsinghua.edu.cn/simple labelimg

打开方式：虚拟环境输入

labelimg

2.建立目录
在data文件夹内新建数据集文件夹并将训练集与验证集的图片和对应txt文件放入images和labes文件夹内。
（data为YOLOv5的文件夹，已经存在。）
3.建立yaml文件，格式如下。（标注两个目标）并将其放到data文件夹下。

train: data/cotton/train/images
val: data/cotton/val/images
nc: 2
names:
  0: 'cotton'
  1: 'zazhi'

4.修改train参数（441行，改为自己的相对地址）–data处；或直接在形参里面修改–data data/名字.yaml

训练即可！！
网上查阅资料整理出来得，不足请指正，一块讨论。
不懂得可私信。无偿的奥~