caffe-ssd训练自己的数据集

上一篇文章记录了caffe-ssd安装过程中的坑和解决办法，https://blog.csdn.net/lukaslong/article/details/81390276

接下来分三部分，分别介绍如何用配置好的caffe-ssd训练VOC数据及自己的数据集

第一部分，训练VOC数据集

1. 预训练模型

caffe-ssd作者提供了预训练模型，下载地址https://gist.github.com/weiliu89/2ed6e13bfd5b57cf81d6

下载完成后保存在 caffe/models/VGGNet 中

2. 下载VOC2007和VOC2012数据集：

在home根目录下创建文件夹data，即/home/×××/data/

cd data

wget http://host.robots.ox.ac.uk/pascal/VOC/voc2012/VOCtrainval_11-May-2012.tar
wget http://host.robots.ox.ac.uk/pascal/VOC/voc2007/VOCtrainval_06-Nov-2007.tar
wget http://host.robots.ox.ac.uk/pascal/VOC/voc2007/VOCtest_06-Nov-2007.tar

tar -xvf VOCtrainval_11-May-2012.tar
tar -xvf VOCtrainval_06-Nov-2007.tar
tar -xvf VOCtest_06-Nov-2007.tar

如果终端下载速度太慢，也可以到相应网站自行下载，然后放到data/文件夹内，进行解压。

可以得到home/.../data/VOCdevkit/文件夹，包含VOC0712、VOC2007、VOC2012三个子文件夹

3. 创建lmdb文件

VOC的数据及标注是不能直接拿来训练的，先要进行格式转换，创建train和test的lmdb文件：

cd caffe

# Create the trainval.txt, test.txt, and test_name_size.txt in data/VOC0712/

./data/VOC0712/create_list.sh

# You can modify the parameters in create_data.sh if needed.
# It will create lmdb files for trainval a