win10 GPU训练yolov3数据集(darknet版)之生成darknet.exe
一、安装
1.安装CUDA和CUDNN
先确定你的显卡是否适宜安装cuda,不能安装cuda,就不用往下看了,浪费时间。
CUDA下载地址 :CUDA.
建议安装CUDA10.0以下的版本
建议全部默认安装就行了,记住所安装的位置,配置环境变量:我的电脑—>属性—>高级系统设置—>环境变量—>系统变量,添加如下两个变量
再在path中新增%CUDA_PATH%\bin
和%CUDA_PATH%\libnvvp
验证环境变量
CUDNN下载地址:CUDNN
要下载与CUDA对应的CUDNN
把下载下来的压缩包中三个文件夹,分别对应名称复制到CUDA安装目录下相对应的文件夹下。最后再在环境变量path中添加%CUDA_PATH%\lib\x64
2.安装VS2017
直接去官网下载VS2017即可。
VS2017官网
3.安装OpenCV
选择一个版本下载,我这边选择的是OpenCV-3.4.1版本
要注意后缀有vc15表示的是能够使用与VS2017
vs14对应VS2015看自己下载的是什么版本的VS。
OpenCV下载网站
二、生成darknet.exe
1.下载darknet
git clone https://github.com/AlexeyAB/darknet
2.VS配置CUDA
在开始菜单中,找到点击Browse CUDA Samples,找到Samples_vs2017.sln点击打开编译。
等待了很久之后,进入界面
直接运行
宜出现的错误是找不到sdk版本,那么就去Visual Studio installer去下载对应的版本就可了。
3.配置OpenCV
进入到克隆下来的文件夹darknet中,再进入build\darknet中,点击darknet.sln文件。
需要重定向,由于之前的版本使用的是VS2015的缘故,所以直接点击确定。
进去之后修改为
在上方点击调试—>darknet属性—>VC++目录—>编辑
包含目录添加(注意是自己的对应目录)
D:\opencv\opencv\build\include\opencv2
D:\opencv\opencv\build\include\opencv
D:\opencv\opencv\build\include
再在库目录中添加
D:\opencv\opencv\build\x64\vc15\lib
再选择链接器—>输入—>附加依赖项,添加这两项。
最好是先去查看opencv\build\x64\vc15\lib中有没有这两个文件。
如果没有就看有没有两个lib所相对应的文件,复制下来修改成这两个名字,再添加进去。
opencv_world3410.lib
opencv_world3410d.lib
右下角点应用,则配置完成。
4.生成darknet.exe
在项目darknet中运行,即可生成darknet.exe
进入darknet\build\darknet\x64中有darknet.exe生成
三、测试
下载https://pjreddie.com/media/files/yolov3.weights
该文件
复制到darknet\build\darknet\x64文件夹中
在此文件夹中打开PowerShell窗口,输入
.\darknet.exe detect cfg/yolov3.cfg yolov3.weights data/dog.jpg
得到以下结果,即为成功。