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点击打开编译。
在这里插入图片描述
等待了很久之后,进入界面
![在这里插入图片描述](https://img-blog.csdnimg.cn/20200423194232497.png?x-oss-process=image/watermark,type_ZmFuZ3po

直接运行
宜出现的错误是找不到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

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

在这里插入图片描述

发布了2 篇原创文章 · 获赞 4 · 访问量 3506

猜你喜欢

转载自blog.csdn.net/qq_42751792/article/details/105703817