目录
- 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数据集