win10 GPU训练yolov3数据集(darknet版)之生成darknet.exe

win10 GPU训练yolov3数据集(darknet版)之生成darknet.exe

  • 一、安装
    • 1.安装CUDA和CUDNN
    • 2.安装VS2017
    • 3.安装OpenCV
  • 二、生成darknet.exe
    • 1.下载darknet
    • 2.VS配置CUDA
    • 3.配置OpenCV
    • 4.生成darknet.exe
  • 三、测试

一、安装

1.安装CUDA和CUDNN

先确定你的显卡是否适宜安装cuda,不能安装cuda,就不用往下看了,浪费时间。
CUDA下载地址 :CUDA.
建议安装CUDA10.0以下的版本
win10 GPU训练yolov3数据集(darknet版)之生成darknet.exe_第1张图片
建议全部默认安装就行了,记住所安装的位置,配置环境变量:我的电脑—>属性—>高级系统设置—>环境变量—>系统变量,添加如下两个变量
在这里插入图片描述
再在path中新增%CUDA_PATH%\bin%CUDA_PATH%\libnvvp
验证环境变量
win10 GPU训练yolov3数据集(darknet版)之生成darknet.exe_第2张图片
CUDNN下载地址:CUDNN
要下载与CUDA对应的CUDNN
win10 GPU训练yolov3数据集(darknet版)之生成darknet.exe_第3张图片

把下载下来的压缩包中三个文件夹,分别对应名称复制到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点击打开编译。
win10 GPU训练yolov3数据集(darknet版)之生成darknet.exe_第4张图片
等待了很久之后,进入界面
win10 GPU训练yolov3数据集(darknet版)之生成darknet.exe_第5张图片

直接运行
宜出现的错误是找不到sdk版本,那么就去Visual Studio installer去下载对应的版本就可了。

3.配置OpenCV

进入到克隆下来的文件夹darknet中,再进入build\darknet中,点击darknet.sln文件。
win10 GPU训练yolov3数据集(darknet版)之生成darknet.exe_第6张图片
需要重定向,由于之前的版本使用的是VS2015的缘故,所以直接点击确定。
进去之后修改为
win10 GPU训练yolov3数据集(darknet版)之生成darknet.exe_第7张图片
在上方点击调试—>darknet属性—>VC++目录—>编辑
包含目录添加(注意是自己的对应目录

D:\opencv\opencv\build\include\opencv2
D:\opencv\opencv\build\include\opencv
D:\opencv\opencv\build\include

win10 GPU训练yolov3数据集(darknet版)之生成darknet.exe_第8张图片
再在库目录中添加

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

得到以下结果,即为成功。

win10 GPU训练yolov3数据集(darknet版)之生成darknet.exe_第9张图片

你可能感兴趣的:(cuda,windows,opencv)