py-faster-rcnn windows 安裝
阿新 • • 發佈:2019-02-11
rbg提供的python版本的faster-rcnn 在ubuntu 下安裝比較容易,跟著敲命令就是了,如果想在window下安裝卻無從下手,搜了幾十上百篇的部落格以後還是沒能成功。能看到的都是過於簡單,幾句話就說完了,採坑無數後想把自己的一些做法寫下來,或許對一些人有用。
安裝環境:win7, python2.7 , cpu 版
1.下載需要的安裝包
2.編譯caffe-rfcn
這裡使用caffe-rfcn而不是官方的caffe是因為在安裝的時候發現這個還是比較靠譜的,沒有報太多的錯。當然這個也和機子的安裝環境有關,遇到問題還要根據具體情況來尋找解決方法。
至於caffe-rfcn 的編譯可以參考官網的教程,也可以用 tina的教程
faster-rcnn 使用到的caffe 和官方的caffe不同的是多了python layer。
3.編譯py-faster-rcnn
在編譯py-faster-rcnn之前把第二步編譯好的caffe-rfcn中的pycaffe中的檔案複製到py-faster-rcnn中的caffe-fast-rcnn資料夾下的python檔案中。
此時把py-faster-rcnn-windows檔案下的lib替換到py-faster-rcnn中的lib
此時安裝lib中的setup.py
python setup.py install
然而這裡會出錯,需要對setup.py進行修改
gcc.exe: error: /Qstd=c99: No such file or directory
error: command 'C:\Anaconda2\Scripts\gcc.exe' failed with exit status 1
解決方法:
把setup.py.中的'/Qstd=c99' 改為 '-std=c99'
更多可以點選這裡
注:使用cuda 還有之執行python setup_cuda.py
4.Test
如果上一步沒有錯那麼到了這一步就是測試一下安裝好了沒有。
執行py-faster-rcnn中tools下的demo.py,在根目錄下執行:
python demo.py --cpu #指定在cpu下執行
這裡可能還會遇到一個報錯
proposal_layer.py 中 pre_nms_topN = cfg[cfg_key].RPN_PRE_NMS_TOP_N, keyerror = 1
解決方法:
#做以下修改
#cfg_key = str(self.phase) # either 'TRAIN' or 'TEST'
cfg_key = str('TRAIN' if self.phase == 0 else 'TEST')
如果遇到更多的報錯參考可以以下部落格:
[找到相應問題]
最後的結果:
注:可以加交流群282617340,尋求幫助或幫助更多有疑問的童鞋。