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