yolov5模型训练及文件类型转换

以VOC数据集为例

在voc文件夹打开终端

python voc_label.py

打开voc2007，打开终端（把生成的labels文件下的txt文件移至图片目录）

mv lables/*.txt JPEGImages/

pycharm打开yolov5文件，复制一份coco128.yaml

终端运行

cp data/coco128.yaml data/garbage.yaml

在复制后的文件修改类别和类别名称

复制models 中的yolov5s.yaml文件

cp models/yolov5s.yaml models/garbage_yolov5.yaml

在复制后的文件中修改类别

训练模型

python train.py --weights yolov5s.pt --cfg models/garbage_yolov5.yaml --data/gatbage.yaml --epoch 10 --batch-size 8

完成后将best.pt文件进行wts转换

wts文件转换

git clone https://github.com/wang-xinyu/tensoorrtx.git

将文件tensorrtx/yolov5/gen_wts.py复制到yolov5文件下

在yolov5文件下打开终端

执行生成yolov5s.wts文件