Single Shot multibox Detector 安装教程

在安装之前说明下我的电脑配置

 Ubuntu14.04

 CUDA7.5

NVIDIA GT750Ti

主要参考链接是 https://github.com/weiliu89/caffe/tree/ssd ，以下是步骤在其基础上做的补充

 1.Get the code.

 git clone https://github.com/weiliu89/caffe.git

 cd caffe 

 git checkout ssd

2.Build the code.

 cp Makefile.config.example Makefile.config

 make -j8

 sudo vi ~/.bashrc   

加入你自己的caffe下的python路径，我的是

 export PYTHONPATH=$PYTHONPATH:/home/xk/caffe/python

 source ~/.bashrc

 make py

 make test -j8 

 make runtest -j

Train参考https://github.com/weiliu89/caffe/tree/ssd注意例子训练是有预训练文件的

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

3.Error Solution

1.error == cudaSuccess (10 vs. 0)  invalid device ordinal ，解决办法：设置好GPU    gpus = "0,1,2,3" 改为gpus = "0"02.error == cudaSuccess (2 vs. 0)  out of memory，解决办法：调小batchsize大小    batch_size = 1    accum_batch_size = 1

2.If you have multiple GPUs installed in your machine, make runtest might fail. If so, try following:

export CUDA_VISIBLE_DEVICES=0;

make runtest -j8

3.Ifyou have error: "Check failed: error == cudaSuccess (10 vs. 0)  invalid device ordinal",first make sure you have the specified GPUs, or try following if you have multiple GPUs:

unset CUDA_VISIBLE_DEVICES