本教程是作者自己参考了一系列资料,以及踩坑之后总结下来的。和大家分享一下。
这次使用的模型是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文件中,需要注意的是,不要包含后缀名。