yolov4-tiny的部署

一、环境配置

见上篇文章 (112条消息) Unet环境配置pytroch框架_风吹吹就过的博客-CSDN博客https://blog.csdn.net/weixin_44751438/article/details/123056809?spm=1001.2014.3001.5501

二、yolov4-tiny的训练及预测

1.数据集的制作

1）进行标注

Win+R启动cmd，在命令提示符内输入以下命令：

labelimg

 注意：将标签文件放入到Annotations文件夹当中，将图片放入到JPEGImages中

2）将Annotations和JPEGImages文件夹复制到 VOCdevkit/VOC2007路径下

3）打开voc_annotation.py文件

首先，修改 annotation_mode = 0 运行，生成下面四个文件

其次，修改 annotation_mode = 2 运行，生成下面两个文件

 4）修改model_data/voc_classes.txt当中的类别即可

 2.训练数据集

运行 train.py文件即可开始训练

三、预测

1.将logs下的权重文件复制到model_data下

 2.修改yolo.py文件中下图所示内容

3.打开 predict.py进行预测