1. 程式人生 > >faster rcnn踩坑記錄

faster rcnn踩坑記錄

1.下載Faster-RCNN原始碼並且進行安裝

 通過git工具直接下載,在linux命令列輸入

# Make sure to clone with --recursive

git clone --recursive https://github.com/rbgirshick/py-faster-rcnn.git

這裡的recursive表示的是遞迴下載的意思,把目錄下的caffe-faster-rcnn現在下來

 2.編譯Cython模組

在Faster-RCNN/lib 目錄下執行一下命令:

cd $FRCN_ROOT/lib
           make

這裡面一定要make一下,,我之前沒make然後花了將近一天在解決這個問題!!!

3.編譯caffe和Pycaffe

開啟caffe-fast-rcnn資料夾看是否有Makefile.config存在

沒有的話拷貝一份

cp Makefile.config.example Makefile.config

首先要對MakeFile.config裡的內容進行修改

將上面兩行註釋取消,這裡面注意自己的python路徑,opencv版本,以及使用的是GPU還是cpu,我使用的是gpu

make

編譯完成 然後 make pycaffe

4.下載事前選練過的Fater R-CNN 的訓練資料

cd $FRCN_ROOT
           ./data/scripts/fetch_faster_rcnn_models.sh

網速太渣,所以我先用百度雲下下來然後解壓拷貝到/media/data1/caius/py-faster-rcnn/data資料夾下

百度雲faster_rcnn_models    https://pan.baidu.com/s/1RTHV7CO6OKNSiShQD0eSwA

5.執行基於python的demo

     cd $FRCN_ROOT
     ./tools/demo.py

 

demo.py會報 no display name and no $DISPLAY environment variable

           查了相關部落格說要新增matplotlib.use('Agg')這句話

在demo.py的最上面加上

import matplotlib                                                                                                         
           matplotlib.use('Agg')

就可以解決這個問題

 

 

參考的部落格:

https://github.com/rbgirshick/py-faster-rcnn

https://blog.csdn.net/u014696921/article/details/52488636

           https://blog.csdn.net/sinat_30372583/article/details/79502078