ICode9

精准搜索请尝试: 精确搜索
首页 > 其他分享> 文章详细

YOLOv4手把手教程!从配置环境,带你跑通代码!

2021-05-18 11:33:46  阅读:316  来源: 互联网

标签:教程 YOLOv4 手把手 data yolov4 CUDA cfg 安装 下载


摘要

YOLOv4新鲜出炉,应该很多人都想尝试一下,但是对于小白,配置环境这一步就劝退了不少人,让我来带着你从配置环境开始,成功运行YOLOv4代码。

1 配置环境

1.1 安装VS

vs下载网址
安装的组件只需要“使用C++的桌面开发”。
我使用的是VS2017(社区),如果你安装的是其他版本应该也问题不大。

1.2 安装OPENCV

下载网站
从里面找到对对应的版本进行下载,下载完成后得到以下文件:
在这里插入图片描述

1.3 安装CUDA

下载地址
先进入NVDIA控制面板:
在这里插入图片描述
进入系统信息,查看你的显卡驱动支持的最高版本的CUDA版本,如果版本过低需要更新显卡(显卡驱动下载,找到你的显卡的型号信息下载下来安装即可):
在这里插入图片描述
选择你想装的版本,我选择的是CUDA10.1:
在这里插入图片描述
选择适合你的操作系统进行下载:
在这里插入图片描述
下载完成后打开,这个位置随便选,因为安装完成后需要删除。在这里插入图片描述
点击ok开始安装。在这里插入图片描述
现在开始正式安装显卡驱动。在这里插入图片描述
使用自定义安装方式:在这里插入图片描述
选择安装内容:在这里插入图片描述
下一步选择默认方式,如果安装成功就进入下一步
环境变量配置:在这里插入图片描述
系统变量添加:
在这里插入图片描述
如果安装时出现这样的问题:在这里插入图片描述
查看是因为那个文件没有安装成功,如果是因为如下内容安装失败:
在这里插入图片描述
在这一步时取消这一项,同时进入我们之前第一步安装的文件夹下,把这些文件复制到 安 装 v s 的 文 件 夹 安装vs的文件夹 安装vs的文件夹\Common7\IDE\VC\VCTargets\BuildCustomizations的路径下
在这里插入图片描述

1.4 安装CUDNN

下载网址
这个安装的时候需要注册一个账号,而且网站有些卡
在这里插入图片描述
选择对应的CUDNN版本:在这里插入图片描述
只需要把下载后的文件安装到当前路径,分别将cuda/include、cuda/lib、cuda/bin三个目录中的内容拷贝到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1对应的include、lib、bin目录下即可。
在这里插入图片描述

1.5 安装cmake

下载地址
点击下载即可,这个下载很慢。
在这里插入图片描述
下载完成后安装得到以下文件:
在这里插入图片描述

2代码下载

2.1 git bash安装

下载地址
下载完成后安装即可,安装时候。记得勾上如下两项即可。

2.2 下载代码

选定要下载的文件夹,右键。
在这里插入图片描述

git clone git://github.com/AlexeyAB/darknet.git

2.3 环境变量配置

在系统变量Path中加入你下载的darknet的位置
在这里插入图片描述

3 运行代码

3.3 使用cmake

在这里插入图片描述
对比以下内容是否相符:
CMAKE部分
在这里插入图片描述
CUDA部分(如果没有这部分可能是CUDA安装失败或者VS的CUDA依赖没装)
在这里插入图片描述
CUDNN部分(如果没有这部分可能是CUDA安装失败或者VS的CUDA依赖没装)
在这里插入图片描述
ENABLE部分,注意开启CUDA
在这里插入图片描述

3.2 编译代码

以此点击生成和打开项目。
在这里插入图片描述
之后会自动打开vs,这时生成文件即可。
在这里插入图片描述
在这里插入图片描述

3.3 运行代码

使用摄像头
在这里插入图片描述
在这里插入图片描述
其他代码:
识别图片(图片放到data目录下)

darknet.exe detect cfg\yolov4.cfg yolov4.weights data\xxx.jpg

打开电脑摄像头识别监控画面(需要提前开启摄像头权限)

darknet.exe detector demo cfg\coco.data cfg\yolov4.cfg yolov4.weights

识别视频(视频放到data目录下)

darknet.exe detector demo cfg\coco.data cfg\yolov4.cfg yolov4.weights .\data\xxx.mp4

后记

跑通代码只是第一步,第二步是用YOLOv4训练我们需要的数据。之后还需要理解算法的原理,最后如果可能再分析一下原码。如果有兴趣想了解原理的可以考虑关注一下博主,应该会在近几个月更新。

标签:教程,YOLOv4,手把手,data,yolov4,CUDA,cfg,安装,下载
来源: https://blog.csdn.net/qq_45662588/article/details/116979782

本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享;
2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关;
3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关;
4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除;
5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

专注分享技术,共同学习,共同进步。侵权联系[81616952@qq.com]

Copyright (C)ICode9.com, All Rights Reserved.

ICode9版权所有