在keras-yolo3使用自己的VOC数据集训练

本教程是作者自己参考了一系列资料，以及踩坑之后总结下来的。和大家分享一下。

 

这次使用的模型是YOLOv3 的keras版本。

链接如下：

https://github.com/qqwweee/keras-yolo3

具体的使用细节，github上都写明了。

我这里是为了检测图像，所示使用，记得下载yolo的权重。

python yolo_video.py --image

就行了

要使用VOC的数据集。我们需要做的是在当前项目的目录下新建VOCdevkit文件夹。

文件的目录如下：

VOCdevkit
    --VOC2007
        --Annotations   用于存档xml文件
        --ImageSets
            --Main 存放训练集和测试集需要用要的文件编号   
        --JPEGImages2  存放需要检测的图片

我们需要在Main目录下，创建两个文件。test.txt和train.txt

这两个文件是给voc_annotation.py调用的，目的是用来生成符合keras-yolo3的数据格式。

如何生成test.txt和train.txt

这两个文件时需要我们自己编写的，我利用的是python脚本将Annotations里面的文件，读取出来随机打乱，然后按照八二开的原则，将文件名字放入train和test文件中，需要注意的是，不要包含后缀名。