Fast R-CNN 1.編譯配置

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張圖,這里貼出其中一張結果的效果圖(很棒喲,兄弟)


demo

如果你的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]

參考

最后編輯于
?著作權歸作者所有,轉載或內容合作請聯(lián)系作者
【社區(qū)內容提示】社區(qū)部分內容疑似由AI輔助生成,瀏覽時請結合常識與多方信息審慎甄別。
平臺聲明:文章內容(如有圖片或視頻亦包括在內)由作者上傳并發(fā)布,文章內容僅代表作者本人觀點,簡書系信息發(fā)布平臺,僅提供信息存儲服務。

相關閱讀更多精彩內容

友情鏈接更多精彩內容