天天看點

py-faster-rcnn windows 安裝

rbg提供的python版本的faster-rcnn 在ubuntu 下安裝比較容易,跟着敲指令就是了,如果想在window下安裝卻無從下手,搜了幾十上百篇的部落格以後還是沒能成功。能看到的都是過于簡單,幾句話就說完了,采坑無數後想把自己的一些做法寫下來,或許對一些人有用。
           

安裝環境:win7, python2.7 , cpu 版

1.下載下傳需要的安裝包

py-faster-rcnn

py-faster-rcnn-windows

caffe-rfcn

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 安裝
py-faster-rcnn windows 安裝

此時把py-faster-rcnn-windows檔案下的lib替換到py-faster-rcnn中的lib

py-faster-rcnn windows 安裝

此時安裝lib中的setup.py

然而這裡會出錯,需要對setup.py進行修改

gcc.exe: error: /Qstd=c99: No such file or directory
error: command 'C:\Anaconda2\Scripts\gcc.exe' failed with exit status 1
           

解決方法:

更多可以點選這裡

如果是protobuf的錯可以參考python protof 在window下的安裝

注:使用cuda 還有之執行python setup_cuda.py

4.Test

如果上一步沒有錯那麼到了這一步就是測試一下安裝好了沒有。

運作py-faster-rcnn中tools下的demo.py,在根目錄下執行:

這裡可能還會遇到一個報錯

proposal_layer.py 中 pre_nms_topN  = cfg[cfg_key].RPN_PRE_NMS_TOP_N, keyerror = 
           

解決方法:

#做以下修改
#cfg_key = str(self.phase) # either 'TRAIN' or 'TEST'
 cfg_key = str('TRAIN' if self.phase ==  else 'TEST')
           

如果遇到更多的報錯參考可以以下部落格:

[找到相應問題]

最後的結果:

(http://blog.csdn.net/chenzhi1992/article/details/52618386)

py-faster-rcnn windows 安裝
py-faster-rcnn windows 安裝
py-faster-rcnn windows 安裝

注:可以加交流群282617340,尋求幫助或幫助更多有疑問的童鞋。

繼續閱讀