admin管理员组

文章数量:1581067

(windows 10 下运行Faster R-CNN(TF)这里写自定义目录标题)
目前深度学习目标检测算法大部分是需要在ubuntu系统下进行编译运行的,但是对于非专业人员来说,linux系统操作起来实在是不方便,本人也是在看了很多论文、视频教程、和其他大神博客之后才在win10下跑通了Faster R-CNN.在这里要向任少卿大神致敬。
废话不多说,直接上干货:
步骤1:配置电脑。首先要保证你的电脑拥有显存大于等于6g的英伟达显卡,我用的卷积网络是VGG16,本人亲测,在win10环境下,GTX 1060 3G 显卡是跑不了Faster R-CNN的,后来换成了GTX 1060 6G,能够跑通demo和train。首先下载cuda_8.0.44_win10.exe 和cudnn-8.0-windows10-x64-v5.1,下载地址https://developer.nvidia/cuda-downloads,先安装cuda,之后将cudnn中bin,include,lib文件里的文件复制替换到cuda的c盘安装目录(C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0)下,然后需要配置环境变量,具体如下:
%PATH%
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\bin;C:\Windows\System32

CUDA_BIN_PATH
%CUDA_PATH%\bin

CUDA_LIB_PATH
%CUDA_PATH%\lib\x64

CUDA_PATH
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0

CUDA_PATH_V7_5

本文标签: WindowsfasterTFCNN