1. 程式人生 > >py-faster-rcnn windows 安裝

py-faster-rcnn windows 安裝

 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檔案中。

複製編譯好的pycaffe

放到py-faster-rcnn中

此時把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,尋求幫助或幫助更多有疑問的童鞋。