【目标检测】基于pytorch框架复现YOLOv4所使用的数据制作脚本

1.数据制作

第一步：收集图片，我们以训练挖掘机和推土机检测摸为例，从网上找了一些图片如下：

我们可以一张一张手动保存图片，也可以使用爬虫爬取，将收集的图片放入JPEGImages文件夹下。

使用labelimg标注工具对收集的数据集进行标注，这一步做目标检测的应该都会，对目标绘制矩形框，并命名：

将每张图片标注得到的xml标签放入Annotations文件夹下。

最后数据存放格式如下，JPEGImages文件夹存放所有图片，Annotations文件夹存放所有标签：

VOCdevkit
    --VOC2007
        --JPEGImages
            --1.jpg
            --2.jpg
        --Annotations
            --1.xml
            --2.xml

2.数据划分为训练集、验证集、测试集

使用1.makeTxt.py脚本将数据划分为训练集、验证集、测试集，1.make