SSD-Pytorch超详细教程

SSD-Pytorch超详细教程

我们使用的是https://github.com/amdegroot/ssd.pytorch该开源代码。

数据集制作

该代码支持两种数据格式，VOC和COCO。我们使用的是VOC。

VOC2007数据格式

VOC2007里包含上述几个文件夹。
Annotations里包含所有图片的xml文件。
jpegimages里面包含所有的图片。
imagesets里面包含三个文件夹，但是检测识别只需要main文件夹，因此我们只新建了main文件夹。


main文件夹主要包含以下几个txt文件，分别用来训练测试和验证。

生成txt代码

import os
import random

trainval_percent = 0.5
train_percent = 0.5
xmlfilepath = 'D:\wxy\ssd.pytorch-master\data\VOCdevkit\VOC2007\Annotations'
txtsavepath = 'D:\wxy\ssd.pytorch-master\data\VOCdevkit\VOC2007\ImageSets\Main'
total_xml =