py-faster-rcnn在windows下安裝


准備好vs2013+anaconda2+好點的顯卡(比如GTX970)

先改造caffe-rfcn以支持cudnn5,得到caffe-rfcn-cudnn5:替換個別代碼文件和代碼;改.props

再用rbg的py-faster-rcnn所依賴的caffe-py-faster-rcnn(這個名字我自己起的,方便區分),里面的SmoothL1Layer的3個文件替換caffe-rfcn-cudnn5的

再編譯py-faster-rcnn的lib,先前裝了vc9for python要卸掉,gpu_nms.cu從github上找。py代碼的param_str_->param_str

成功后訓練需要~1800M顯存,訓練和測試同時執行需要~2600M顯存


免責聲明!

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



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