准备数据集
mkdir data
cd data
下载VOC2007与2012数据集
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
创建lmdb格式的数据:
cd caffe
./data/VOC0712/create_list.sh
./data/VOC0712/create_data.sh
下载预训练模型,放在caffe/models/VGGNet/:
链接: https://pan.baidu.com/s/1jjxB9XPOvuXZUJSTovCQHg 密码: ypji
训练
python examples/ssd/ssd_pascal.py
也可以下载官方训练好的模型
http://www.cs.unc.edu/%7Ewliu/projects/SSD/models_VGGNet_VOC0712_SSD_300x300.tar.gz
解压后,直接把model文件夹下的VGGNet复制粘贴到caffe/models/下就可以了
报错1
ImportError: No module named caffe
或者ImportError: No module named _caffe
这是因为你没有把caffe加入环境变量,加入即可
##测试
测试图像
python examples/ssd/score_ssd_pascal.py
报错
1 ./data/VOC0712/create_list.sh
后
Create list for VOC2007 trainval...
cp: 无法获取'/home/ye/data/VOCdevkit//VOC2007/ImageSets/Main/trainval.txt' 的文件状态(stat): 没有那个文件或目录
sed: 无法读取 /home/ye/caffe/data/VOC0712/trainval_img.txt: 没有那个文件或目录
solution:
修改create_list.sh文件里data_root_dir,将其改为绝对路径=”/home/ye/caffe/data/VOCdevkit”
./data/VOC0712/create_list.sh
也类似