用YOLOv5模型训练自己的数据集

1.首先准备好yolo格式的数据集，这是已经转化好的：

 2.数据集的配置文件，直接用output中生成的data.yaml就可以（也可以自己改），names就是标注的种类

3.修改yolov5s.yaml，nc改成自己数据集类别的个数

 

 4.修改train.py(或直接在命令行指定对应超参数)根据自己的条件修改，显卡内存小的话，把batch_size调小一点

5.训练数据集

可以可视化训练过程：tensorboard --logdir=runs

 6.评估训练的模型

训练的模型在这里