MobileNet-SSD训练

原创

于 2018-09-28 18:11:00 发布 · 2.2k 阅读

CC 4.0 BY-SA版权

本文档详细介绍了如何在Caffe环境下训练MobileNet-SSD模型，包括下载模型代码、准备VOC数据集、创建数据列表、生成训练与测试文件、修改配置文件、训练模型以及进行图片和视频的测试。通过遵循这些步骤，读者可以成功地训练自己的MobileNet-SSD模型。

在caffe-ssd环境搭建好的情况下

1.下载MobileNet-SSD
cd /home/**/caffe/examples/
git clone https://github.com/chuanqi305/MobileNet-SSD

2.准备数据集，我直接下载的VOC数据集，也可以仿制VOC，使用自己的数据集
a.直接用VOC数据集
cd /home/
mkdir 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
解压后可以看到一个VOCdevkit的文件夹
cd /home/**/caffe
执行：
./data/VOC0712/create_list.sh
./data/VOC0712/create_data.sh

再进入到home下的data文件夹
cd VOCdevkit
cd VOC0712
可以看到一个lmdb文件夹
cd lmdb
可以看到两个文件夹，分别是VOC0712_test_lmdb VOC0712_trainval_lmdb
将这两个文件夹直接拷贝到 MobileNet-SSD目录下
并将名称改为trainval_lmdb 和 test_lmdb
#######
制作labelmap.protot