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

摘要

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

1 配置环境

1.1 安装VS

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

1.2 安装OPENCV

下载网站
从里面找到对对应的版本进行下载，下载完成后得到以下文件：

将文件提取到指定位置。

然后我们进行环境变量配置，在系统变量中添加以下数据：

再进入系统变量里的Path，添加如下数据。

1.3 安装CUDA

下载地址
先进入NVDIA控制面板：

进入系统信息，查看你的显卡驱动支持的最高版本的CUDA版本，如果版本过低需要更新显卡（显卡驱动下载，找到你的显卡的型号信息下载下来安装即可）：

选择你想装的版本，我选择的是CUDA10.1：

选择适合你的操作系统进行下载：

下载完成后打开，这个位置随便选，因为安装完成后需要删除。

点击ok开始安装。

现在开始正式安装显卡驱动。

使用自定义安装方式：

选择安装内容：

下一步选择默认方式，如果安装成功就进入下一步
环境变量配置：

系统变量添加：

如果安装时出现这样的问题：

查看是因为那个文件没有安装成功，如果是因为如下内容安装失败：


在这一步时取消这一项，同时进入我们之前第一步安装的文件夹下，把这些文件复制到$安装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

下载地址
点击下载即可，这个下载很慢。

下载完成后安装得到以下文件：

进入bin文件夹，这个cmake-gui就是我们要用程序。

2代码下载

2.1 git bash安装

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

2.2 下载代码

选定要下载的文件夹，右键。

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

2.3 环境变量配置

在系统变量Path中加入你下载的darknet的位置

3 运行代码

3.3 使用cmake

打开cmake指定路径

点击左下角按钮。

在这两个地方勾选，便于修改和观察

对比以下内容是否相符：
CMAKE部分

CUDA部分（如果没有这部分可能是CUDA安装失败或者VS的CUDA依赖没装）

CUDNN部分（如果没有这部分可能是CUDA安装失败或者VS的CUDA依赖没装）

ENABLE部分，注意开启CUDA

INSTALL部分

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训练我们需要的数据。之后还需要理解算法的原理，最后如果可能再分析一下原码。如果有兴趣想了解原理的可以考虑关注一下博主，应该会在近几个月更新。