Windows下編譯fast rcnn

查志強發表於2015-09-08

【原文:http://blog.csdn.net/happynear/article/details/46822109

一、準備

  1. caffe-windows:https://github.com/happynear/caffe-windows。如果你的caffe-windows版本是2015/07/09之前下載的,請重新下載並重新編譯。重新編譯時請首先編譯python介面,因為馬上就要用到。主程式和matlab介面可以有空的時候再編譯。

  2. fast r-cnn:https://github.com/rbgirshick/fast-rcnn

  3. 下載fast r-cnn所需的模型檔案。原始檔案都在伯克利大學的伺服器上,為了不給伯克利伺服器增加負擔,我把檔案都上傳到百度雲上了[doge]:http://pan.baidu.com/s/1kTzQYgR。下載完成後,請將3個資料夾放置在fast_rcnn_root/data中。

  4. 按照這篇部落格的說明安裝protobuf的python介面。

  5. 開啟cmd,輸入pip install cython 和 pip install easydict來安裝cython和easydict,如果你使用的是Anaconda,cython預設是安裝上了的。 
    如果提示安裝失敗,可能是由於你的pip的版本過低,請先使用 
    pip install pip --upgrade 
    升級你的pip。

二、編譯步驟

  1. caffe_windows_root/python目錄,複製到fast_rcnn_root/caffe-fast-rcnn中。

  2. 用文字編輯器開啟fast_rcnn_root/lib/utils/nms.pyx,將第25行的np.int_t修改為np.intp_t

  3. 用文字編輯器開啟fast_rcnn_root/lib/setup.py,將第18行和23行的 
    "-Wno-cpp", "-Wno-unused-function"指令刪除,只留下空的中括號[]即可。

  4. 開啟cmd,定位至fast_rcnn_root/lib目錄中,執行python setup.py install。 
    如果提示Unable to find vcvarsall.bat的話,請輸入以下指令: 
    VS2012:SET VS90COMNTOOLS=%VS110COMNTOOLS% 
    VS2013:SET VS90COMNTOOLS=%VS120COMNTOOLS% 
    setup.py安裝完成後,到python_root/Lib/site-packages/utils中可以找到兩個檔案cython_bbox.pydcython_nms.pyd,把這兩個檔案複製到fast_rcnn_root/lib/utils中。

  5. 定位至fast_rcnn_root,執行python tools/demo.py,等待demo的結果吧! 
    fast rcnn result

版權宣告:本文為博主原創文章,未經博主允許不得轉載。


相關文章