上一章已经详细的讲解了VOC转换成YOLO的全过程,这里不在赘述
在训练数据集之前,创建自己的数据集yaml文件,我这里以yolov7为例,在data的文件夹下创建mydata.yaml文件
在文件中添加自己数据集的信息,代码如下
train: 你自己的文件夹/train.txt
val: 你自己的文件夹/val.txt
test: 你自己的文件夹D/test.txt
nc: 4 #类别数量,你自己有几类写几类
names: ["","","",""] #类别的种类
配置完成后,就可以开始训练了
在Terminal中,输入
python train.py --img 640 --batch 2 --epoch 1 --data data/myata.yaml --cfg cfg/training/yolov7.yaml --weights weights/yolov7.pt --device '0' # batch和epoch按你自己的情况来
其中的weights中yolov7.pt,需要到github上下载,
其中运行代码的参数,网上有很多教程,可以自行搜索看一下。