windows下配置Faster-RCNN


mark一個

http://yun.baidu.com/share/link?shareid=1018944597&uk=1543560377

 http://blog.csdn.net/sinat_30071459/article/details/50546891

配置環境: Windows10x64 Matlab2015Ra VS2013 Opencv2.4.11 CUDA7.5 GTX950M

  1. CUDA7.5安裝 
    因為Cuda7.5做了很大的優化改進,而且對win10支持較好,所以這里選擇安裝Cuda7.5,具體過程 參考這里。但是由於作者Matlab 版本的Faster r-CNN的cnn庫是在Cuda6.5下編譯的,所以這里會出現問題,要么需要自己在Cuda7.5下編譯,要么可以在這里下載。

  2. 下載 FasterR-CNN程序包 
    在作者ShaoqingRen的github上下載Matlab版本的faster-cnn:鏈接。然后解壓到本地。

  3. 編譯external/caffe庫 
    如果你的電腦安裝的是CUDA6.5, 那么可以直接運行fetch_data/fetch_caffe_mex_windows_vs2013_cuda65.m。 
    如果你是像本人那樣安裝了CUDA7.5, 那么你可以在這里下載編譯好的庫,直接解壓到作者代碼的根目錄下。

  4. 生成nms mex文件 
    運行文件根目錄下的:faster_rcnn_build.m 
    注意:這里運行的時候會經常報錯,是需要將functions/nms/nvmex.m下的Cuda_path改成你電腦的CUDA安裝路徑。

  5. 設置相關函數路徑 
    運行startup.m文件,這只函數運行的相關路徑。

  6. 下載CNN的models 
    要么運行作者代碼包里的:fetch_data/fetch_faster_rcnn_final_model.m 自動下載並保存在output文件夾里面。 
    或者自己在百度雲里下載,並解壓到output文件夾里。

  7. 測試運行demo 
    如果前面一切ok沒有問題的話,這里將experiments/script_faster_rcnn_demo.m 拷貝到根目錄下,然后運行就可以看到結果了。 
    注意:由於VGG16模型太大了,所以一般電腦運行起來會出現matlab奔潰,所以這里要么將 
    opts.use_gpu = false;在cpu下運行。 
    或者使用ZF模型(比VGG16簡單,准確度降低):

    model_dir= fullfile(pwd, 'output', 'faster_rcnn_final', 'faster_rcnn_VOC0712_ZF');

這樣就可以看到測試圖片的運行結果和運行時間了。 


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM