1. 程式人生 > >深度學習Caffe實戰筆記(18)Windows平臺 Faster-RCNN 環境配置

深度學習Caffe實戰筆記(18)Windows平臺 Faster-RCNN 環境配置

好久不寫部落格了,因為前一段時間博主身體抱恙,感觸頗多。。。。。。都說windows平臺做caffe和tensorflow坑多,博主沒有太多優點,唯獨的一個優點就是不服輸,Windows平臺雖然坑多,但是填坑也是蠻有意思的一件事情。另外,在Siamese網路訓練自己的資料那個部落格,經過高人點評,有些不太完善的地方,具體的資訊請看那篇部落格的一個長評論。

勞動節的時候幫一個師妹配置caffe,她說要做RCNN,caffe雖然配置成功,但是我對RCNN一點都不懂,所以也嘗試先把RCNN配置成功,然後再訓練一點自己的資料,正好接上上學期沒有做完的工作。廢話扯的有點多。。。。。。

博主用的WIN7+cuda7.5+faster-RCNN+vs2013(vs2010)+matlab2014a。因為對MATLAB更熟悉一些,所以就只配置了MATLAB版本的Faster-RCNN
1、安裝VS2013


這個我就不多說了吧
2、安裝MATLAB
參考網上教程說要先安裝VS2013,再安裝MATLAB,如果先安裝MATLAB會出現找不到vs的情況,這個博主沒有試驗過,不知道是不是這個道理。
3、下載Faster-RCNN
下載地址 https://github.com/ShaoqingRen/faster_rcnn 這個是MATLAB版本的,如果要下載python版本的請前往https://github.com/rbgirshick/py-faster-rcnn python版本的配置方法請參考其他高人部落格。
把下載好的faster_rcnn-master解壓到硬碟上。
4、安裝cuda

cuda版本很重要,不同的版本會出現不同的錯誤,下載的faster-rcnn是cuda6.5版本的,所以最好下載cuda6.5版本安裝,博主用的是7.5,如果是別的版本的就要重新編譯這些檔案這裡寫圖片描述

我把7.5版本編譯好的external檔案放在http://pan.baidu.com/s/1gfegVMr,如果需要大家自行下載,替換原來的external資料夾。安裝好cuda是需要配置環境變數的,環境變數的配置方法參照 http://blog.csdn.net/u011314529/article/details/51505029
4、在MATLAB中編譯
在MATLAB命令列視窗輸入 mbuild -setup,選擇編譯器,因為博主安裝了vs2010和vs2013兩個,所以會出現多個編譯器。
這裡寫圖片描述

如果是cuda6.5版本,就執行

fetch_data/fetch_caffe_mex_windows_vs2013_cuda65.m 

如果是7.5就把網盤裡的資料夾放在faster-rcnn根目錄下即可。

5、測試
首先開啟MATLAB,執行
faster_rcnn-master\faster_rcnn_build.m。
如果提示:Link:找不到****.o* 的錯誤,就把faster_rcnn-master\faster_rcnn-master\functions\nms資料夾下的nvmes.m裡的
這裡寫圖片描述

把這個路徑改成你安裝VS2013的路徑就可以了。
然後執行faster_rcnn-master\startup.m
再執行faster_rcnn-master\fetch_data\fetch_faster_rcnn_final_model.m 下載訓練好的模型,這個模型762M,不想下載的同學可以從http://pan.baidu.com/s/1c1rJ58 我分享的網盤裡下載。
修改faster_rcnn-master\experiments\script_faster_rcnn_demo.m的model_dir為你下載的模型,然後執行。

結果
這裡寫圖片描述

這裡寫圖片描述

用自己筆記本跑gpu超級慢有木有!