1.首先准备好yolo格式的数据集,这是已经转化好的:
2.数据集的配置文件,直接用output中生成的data.yaml就可以(也可以自己改),names就是标注的种类
3.修改yolov5s.yaml,nc改成自己数据集类别的个数
4.修改train.py(或直接在命令行指定对应超参数)根据自己的条件修改,显卡内存小的话,把batch_size调小一点
5.训练数据集
可以可视化训练过程:tensorboard --logdir=runs
6.评估训练的模型
训练的模型在这里