最近心血来潮想学一下YOLOv3,因而就去网上看了YOLOv3在win10下的配置教程。在配置过程当中塌坑无数,花了不少时间和精力,因此我想就此写一篇博客来介绍在在win10+vs2015的环境下如何配置YOLOv3。git
尽可能安装vs2015github
2.1肯定本身电脑对应的CUDA版本号shell
2.1.1打开控制面板windows
2.1.2进入NVIDIA控制面板工具
2.1.3选择系统信息
测试
2.1.4点击组件,便可看到本身电脑适应的CUDA的版本号ui
2.2进入官网https://developer.nvidia.com/cuda-downloads ,按照本身的系统选择合适的版本。spa
CUDNN的版本须要与CUDA版本对应,下载的时候根据选定的CUDA版本进行选择便可。.net
下载连接: https://developer.nvidia.com/cudnn命令行
下载以后把对应文件夹的文件直接放入CUDA的对应文件夹就能够了
即
将D:\安装包\cudnn-9.1-windows10-x64-v7.1\cuda\bin\cudnn64_7.dll 复制到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.1\bin
将 D:\安装包\cudnn-9.1-windows10-x64-v7.1\cudacuda\ include\cudnn.h 复制到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.1\include
将D:\安装包\cudnn-9.1-windows10-x64-v7.1\cudacuda\lib\x64\cudnn.lib复制到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.1\lib\x64
复制完以后要去检查一下cuda和cudnn的环境变量 ,务必要保证这些环境变量都有要否则最后在测试时会报错
下载地址:https://sourceforge.net/projects/opencvlibrary/files/opencvwin/3.4.0/opencv-3.4.0-vc14_vc15.exe/download
连接: https://pan.baidu.com/s/1yACjz1AiDWtkUzxZHqCH0w 提取码: scxv
接下来将opencv配置到环境变量
下载地址:https://github.com/AlexeyAB/darknet
(若是会使用git,则直接git clone https://github.com/AlexeyAB/darknet)
5.1修改darknet.vcxproj文件
进入darknet\build\darknet目录,用记事本打开darknet.vcxproj,将全部CUDA 10.0修改成本身对应的CUDA版本,本篇使用CUDA 9.1,故改成9.1。(共有两处须要修改)
5.2打开darknet.sln
5.2.1将项目修改成Release x64
5.2.2配置包含目录 + 库目录 + 连接器
包含目录:在darknet项目上点击鼠标右键->属性,弹出以下界面:而后VC++目录-->包含目录-->编辑
添加的目录:...\opencv\build\include(...表明opencv的安装路径,下文同理)
...\opencv\build\include\opencv
...\opencv\build\include\opencv2
库目录:方法与包含目录相似,添加的目录为:...\opencv\build\x64\vc14\lib
连接器:添加目录...\opencv\build\x64\vc14\lib下库的名字:opencv_world340.lib
5.2.3右键darknet,选择属性,进入CUDA C/C++中的Code Generation,删掉compute_75,sm_75(此步极为关键,要否则必出错)
5.3右键darknet,选择生成
编译成功,会在darknet\build\darknet\x64下获得darknet.exe
当你看到darknet.exe时,那么就恭喜你配置成功过了!
6.1下载yolov3.weights
官网:https://pjreddie.com/media/files/yolov3.weights
下载速度有点慢
也能够用个人连接 连接: https://pan.baidu.com/s/1FSkXMGqCeqDFpeDxJzb3sA 提取码: 1ajt
下载完成后把它放到darknet\build\darknet\x64下
6.2测试
在darknet\build\darknet\x6下按住shift键,点击鼠标右键选择“在此处打开Powershell 窗口(s)”
在命令行输入 .\darknet detect cfg/yolov3.cfg yolov3.weights data/dog.jpg
若是看到如下结果,说明你成功的配好了YOLOv3