about
近期在做一個Object Detective的項目,這里用的是Caffe,如果同學你還沒裝上Caffe的話,可以參考我之前寫的文章:
Ubuntu 14.04LTS + Caffe+ CUDA8.0 + OpenCV 3.1.0 + cuDNN 8.0
在Caffe搭好了之后,再來看這篇。
Installation
clone fast-rcnn
這里,我們先下載Ross Girshick大神在Github上的一個庫。
# Make sure to clone with --recursive
$ git clone --recursive https://github.com/rbgirshick/fast-rcnn.git
如大神所言,記得加上--recursive
Build the Cython modules
從下文開始,我將用$FRCN_ROOT來代表上面git clone下來的文件目錄路徑。如直接放在home下,$FAST_ROOT被替換為/home/cap/fast-rcnn(注意,此處的cap是我的用戶名,應該換成你自己的)
$ cd $FRCN_ROOT/lib
$ make
Build Caffe and pycaffe
在編譯之前,先來做下準備工作。
把$FRCN_ROOT/caffe-fast-rcnn目錄下的Makefile.config.example文件復制一份,重命名為Makefile.config。
修改Makefile.config的相關項
...
USE_CUDNN = 1 //此項默認是關閉的,若你安裝了cudnn,則把前面的#去掉,讓其支持cudnn.
WITH_PYTHON_LAYER = 1 //這項默認也是關閉的,應該打開,去掉前面的#
USE_PKG_CONFIG = 1 //此項默認也是關閉的,應該打開,去掉前面的#
# CPU_ONLY := 1 //此項默認是關閉的,如果你的機器不支持gpu計算,則將其打開,去掉前面的注釋
CUDA_DIR := /usr/local/cuda //注意此項,應該和你安裝的cuda的路徑相同
在完成了前面的工作后,開始編譯pycaffe。
$ cd $FRCN_ROOT/caffe-fast-rcnn
$ make -j8 && make pycaffe
- 編譯pycaffe時,我的機器報錯了
AR -o .build_release/lib/libcaffe.a
LD -o .build_release/lib/libcaffe.so
/usr/bin/ld: cannot find -lippicv
collect2: error: ld returned 1 exit status
make: *** [.build_release/lib/libcaffe.so] Error 1
- 解決方法:
把OpenCV下的libippicv.a文件復制到/usr/local/lib目錄下,然后重新編譯pycaffe。具體操作如下:
復制libippicv.a文件
$ cd /usr/local/opencv-3.1.0/3rdparty/ippicv/unpack/ippicv_lnx/lib/intel64
$ sudo cp libippicv.a /usr/local/lib
然后重新編譯pycaffe
$ cd $FRCN_ROOT/caffe-fast-rcnn
$ make clean #清除前面的
$ make -j8 && make pycaffe
Download pre-computed Fast R-CNN detectors
此處是下載一些訓練模型,在這篇博客里,我們先用作者已經訓練好的caffe模型,看下fast-rcnn的效果:
$ cd $FRCN_ROOT
$ ./data/scripts/fetch_fast_rcnn_models.sh
下載完后,在$FRCN_ROOT/data/fast_rcnn_models下有三個模型,分別為
.
├── caffenet_fast_rcnn_iter_40000.caffemodel
├── vgg16_fast_rcnn_iter_40000.caffemodel
└── vgg_cnn_m_1024_fast_rcnn_iter_40000.caffemodel
在$FRCN_ROOT/data/scripts下,有三個shell(即.sh)文件
.
├── fetch_fast_rcnn_models.sh
├── fetch_imagenet_models.sh
└── fetch_selective_search_data.sh
此處引用樓燚航同學的介紹:
第一是作者訓練好的fast_rcnn模型,第二個是imagenet_model上預訓練好的模型,第三個對應著的是作者基于Pascal VOC數據集提取的selective_search預選框。如果想要看一下fast rcnn的效果,可以直接加載Ross訓練好的fast_rcnn模型,如果要自己訓練的話,記得加載imagenet模型
run the demo
在開始前,你可能需要先安裝相關的Python module
$ sudo pip2 install easydict
開始跑動demo
cd $FRCN_ROOT
./tools/demo.py
運行結果:
結果跑出了3張圖,這里貼出其中一張結果的效果圖(很棒喲,兄弟)

如果你的gpu性能不加,顯存小于3G,那么可能出現(xiàn)類似如下的錯誤
Loaded network /home/hadoop/fast-rcnn/data/fast_rcnn_models/vgg16_fast_rcnn_iter_40000.caffemodel
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
Demo for data/demo/000004.jpg
F0314 16:28:23.588573 5987 syncedmem.cpp:51] Check failed: error == cudaSuccess (2 vs. 0) out of memory
*** Check failure stack trace: ***
Aborted (core dumped)
- 解決辦法:
使用較小的網絡模型
$ cd $FRCN_ROOT
$ ./tools/demo.py --net caffenet
- 如果是你的機器不支持GPU,那么記得在前面的Makefile.config文件去掉 # CPU_ONLY := 1 前面的#,使用CPU計算,然后此處跑動demo時,敲入
$ cd $FRCN_ROOT
$ ./tools/demo.py --cpu
- 想了解更多的操作,敲入
$ cd $FRCN_ROOT
$ ./tools/demo.py -h
usage: demo.py [-h] [--gpu GPU_ID] [--cpu]
[--net {vgg16,caffenet,vgg_cnn_m_1024}]
Train a Fast R-CNN network
optional arguments:
-h, --help show this help message and exit
--gpu GPU_ID GPU device id to use [0]
--cpu Use CPU mode (overrides --gpu)
--net {vgg16,caffenet,vgg_cnn_m_1024}
Network to use [vgg16]
參考
Ross Girshick大神在Github上的搭建說明
dengshuai_super同學的/usr/bin/ld: 找不到 -lippicv(ubuntu 16.04 LTS 安裝opencv-3.1.0,含完整安裝步驟)中的解決方案三