ssd.pytorch指南（四）——制作PASCAL VOC2007数据集

目录

• ssd.pytorch指南（一）——序言
• ssd.pytorch指南（二）——平台搭建
• ssd.pytorch指南（三）——训练模型
• ssd.pytorch指南（四）——制作PASCAL VOC2007数据集
• ssd.pytorch指南（五）——训练自己的数据集

一、需改数据集代码
　　如第二章提到，官方的训练集使用了VOC0712trainval（2007+2012），测试集用了VOC2007test。那么实际上，我们不需要制作两个训练集，只需让模型读取一个训练集即可。
　　找到ssd_pytorch/data/voc0712.py第98行

def __init__(self, root,
             image_sets=[('2007', 'trainval'), ('2012', 'trainval')],
             transform=None, target_transform=VOCAnnotationTransform(),
             dataset_name='VOC0712'):

将2012删掉，改为

def __init__(self, root,
             image_sets=[('2007', 'trainval')],
             transform=None, target_transform=VOCAnnotationTransform(),
             dataset_name='VOC0712'):

二、制作数据集
这一部分的内容和我之前的一篇文章完全相同，就不重复劳动了。
tf-faster-rcnn指南（四）——制作PASCAL VOC2007数据集