Windows下Faster R-CNN 配置/Matlab版本编译
配置環(huán)境: Windows10x64 Matlab2015Ra VS2013 Opencv2.4.11 CUDA7.5 GTX950M
CUDA7.5安裝
因?yàn)镃uda7.5做了很大的優(yōu)化改進(jìn),而且對(duì)win10支持較好,所以這里選擇安裝Cuda7.5,具體過(guò)程 參考這里。但是由于作者M(jìn)atlab 版本的Faster r-CNN的cnn庫(kù)是在Cuda6.5下編譯的,所以這里會(huì)出現(xiàn)問(wèn)題,要么需要自己在Cuda7.5下編譯,要么可以在這里下載。
下載 FasterR-CNN程序包
在作者ShaoqingRen的github上下載Matlab版本的faster-cnn:鏈接。然后解壓到本地。
編譯external/caffe庫(kù)
如果你的電腦安裝的是CUDA6.5, 那么可以直接運(yùn)行fetch_data/fetch_caffe_mex_windows_vs2013_cuda65.m。
如果你是像本人那樣安裝了CUDA7.5, 那么你可以在這里下載編譯好的庫(kù),直接解壓到作者代碼的根目錄下。
生成nms mex文件
運(yùn)行文件根目錄下的:faster_rcnn_build.m
注意:這里運(yùn)行的時(shí)候會(huì)經(jīng)常報(bào)錯(cuò),是需要將functions/nms/nvmex.m下的Cuda_path改成你電腦的CUDA安裝路徑。
設(shè)置相關(guān)函數(shù)路徑
運(yùn)行startup.m文件,這只函數(shù)運(yùn)行的相關(guān)路徑。
下載CNN的models
要么運(yùn)行作者代碼包里的:fetch_data/fetch_faster_rcnn_final_model.m 自動(dòng)下載并保存在output文件夾里面。
或者自己在百度云里下載,并解壓到output文件夾里。
測(cè)試運(yùn)行demo
如果前面一切ok沒(méi)有問(wèn)題的話(huà),這里將experiments/script_faster_rcnn_demo.m 拷貝到根目錄下,然后運(yùn)行就可以看到結(jié)果了。
注意:由于VGG16模型太大了,所以一般電腦運(yùn)行起來(lái)會(huì)出現(xiàn)matlab奔潰,所以這里要么將
opts.use_gpu = false;在cpu下運(yùn)行。
或者使用ZF模型(比VGG16簡(jiǎn)單,準(zhǔn)確度降低):
- 1
這樣就可以看到測(cè)試圖片的運(yùn)行結(jié)果和運(yùn)行時(shí)間了。
關(guān)于‘invalid Mex File’幾點(diǎn)補(bǔ)充
一般是由于系統(tǒng)找不到庫(kù)文件的原因引起的。1. 檢查caffe_faster_rcnn下面的dll庫(kù)是否完整,不完整就重新拷過(guò)來(lái);2. 檢查cuda, opencv的安裝是否完成;3. 檢查系統(tǒng)環(huán)境變量,尤其是在Path中添加cuda, opencv的路徑。by Dr. Sword.
總結(jié)
以上是生活随笔為你收集整理的Windows下Faster R-CNN 配置/Matlab版本编译的全部?jī)?nèi)容,希望文章能夠幫你解決所遇到的問(wèn)題。
- 上一篇: 【caffe-MATLAB】自己编译fa
- 下一篇: python学习网址