原

Caffe深度学习入门——配置caffe-SSD详细步骤以及填坑笔记

2018年01月19日 14:58:22 阅读数：2840 标签：深度学习更多

个人分类： caffe

主要参考 https://github.com/weiliu89/caffe/tree/ssd

获取SSD的代码，下载完成后有一个caffe文件夹

git clone https://github.com/weiliu89/caffe.git
cd caffe
git checkout ssd
      
      
       
       1
       
       2
       
       3

cd caffe
cp Makefile.config.example Makefile.config
      
      
       
       1
       
       2

编译caffe

make all  -j8  //-j16根据本机的处理器配置，16是16核处理器的意思
make pycaffe -j8
make test -j8
make runtest -j8（这一步不是必须的）
      
      
       
       1
       
       2
       
       3
       
       4

下载预训练模型链接：链接：https://pan.baidu.com/s/1o9c0GGU 密码：fucx，将它放入caffe/models/VGGNet/目录下

下载VOC2007和VOC2012数据集，放到/caffe/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
      
      
       
       1
       
       2
       
       3
       
       4
       
       5
       
       6
       
       7

将图片转化为LMDB文件，用于训练,在caffe目录下

cd ..
cd caffe
./data/VOC0712/create_list.sh  
./data/VOC0712/create_data.sh
      
      
       
       1
       
       2
       
       3
       
       4

训练模型
在下载的caffe根目录执行如下命令训练，在examples/ssd下存在几个.py文件，训练的时间较长。

python examples/ssd/ssd_pascal.py
      
      
       
       1

实验效果
（1）在图片测试集上测试

python examples/ssd/score_ssd_pascal.py
      
      
       
       1

2）在视频上测试

   python examples/ssd/ssd_pascal_video.py 
      
      
       
       1

3）在摄像头上测试

   python examples/ssd/ssd_pascal_webcam.py
      
      
       
       1

报错问题解决以及填坑笔记：

1. 报错:no module named caffe等错误

这是由于caffe的Python环境变量未配置好，可按照下面方法解决：

确保先重新make下pycaffe,然后将python的路径配置到环境变量中
make pycaffe -j8
vim ~/.bashrc
加入 export PYTHONPATH=/SSD所在目录/caffe/python
source ~/.bashrc

2. 配置SSD-caffe测试时出现“Check failed: error == cudaSuccess (10 vs. 0) invalid device ordinal”解决

运行 python examples/ssd/ssd_pascal.py 时报错

解决方案：这是由于GPU数量不匹配造成的，如果训练自己的数据，那么我们只需要将solver.prototxt文件中的device_id项改为自己的GPU块数，一块就是0，两块就是1，以此类推。

但是SSD配置时的例子是将训练语句整合成一个python文件ssd_pascal.py，所以需要改此代码。相关配置训练方法请参看转载博文：http://blog.csdn.net/xunan003/article/details/78427446

解决方法：将ssd_pascal.py文件中第332行gpus = “0，1，2，3”的GPU选择改为gpus = “0”，后面的1，2，3都删掉即可。再次训练即可。

当然，由于博主只有一块GPU且电脑运行内存有限，还需要将ssd_pascal.py文件中的337行batch_size = 32和338行accum_batch_size = 32都改小一倍，即更改批量大小，不然会出现“Check failed: error == cudaSuccess (2 vs. 0) invalid …”的错误。

3. 配置SSD-caffe出现“ AttributeError: ‘module’ object has no attribute ‘LabelMap’”解决

这是由于caffe的Python环境变量未配置好，可按照下面方法解决：
vim ~/.bashrc
加入 export PYTHONPATH=/SSD所在目录/caffe/python
source ~/.bashrc