vs2015+opencv440实现YOLOv4,并训练自己的数据集(完全零基础)
最近因为毕设需要,在研究深度学习目标检测的一些算法,具体算法的对比和特点就不详细说了,总之最终确定使用YOLOv4算法。作为深度学习小白,从零开始记录一下自己实现该算法的过程,并且方便自己将来使用。
目前网络上也有很多关于Windows实现YOLOv4的教程,但是总感觉很凌乱,需要不断对比才能找对真正可行的方法,因为本文是一个总结文,希望能给之后学习的人一些帮助。文中用到了一些其他人的程序和图片,我会全部标注原文链接,如有冒犯侵权请原作者联系我删除。
一、环境介绍
对于从零开始的小白,我们需要安装以下软件或环境
VS2015 ,VS2015是我们用到的编译器,需要安装C++和python环境,python只有在我们训练自己的数据集时才用到;
opencv440,opencv440是我们用到的图像处理库,opencv从440开始支持了YOLOv4,所以不要下载更低版本;
NVIDIA驱动程序 ,能够让我们更好地利用自己的显卡;
CUDA10.2 ,CUDA是NVIDIA推出的计算平台,能够基于NVIDIA显卡架构使用GPU进行复杂的图像运算;
cuDNN,cuDNN是专门用于CUDA的神经网络加速包,与CUDA配合使用 ;
YOLOv4相关配置文件
二、环境的具体搭建
VS2015
可以直接到官网 vs下载地址下载最新的2019社区版(社区版是可以免费试用的),其实不同版本差异并不大,都可以使用。发现自己没有网盘会员上传不了vs2015安装包,要是真的想用2015读者可以自行查找一下。
下载完成以后,点击vs_community.exe程序安装。
在选择功能中选择 C++和python进行安装。接下来耐心等待到安装完成即可。
opencv440
链接:https://pan.baidu.com/s/1eUb8oJJS22YwXzWsySki1A
提取码:qdpt
下载完成后,进行安装,具体的安装教程可以参考另外一位博主的文章,下面给出链接。opencv安装与环境配置
NVIDIA驱动程序+CUDA+cuDNN
CUDA与cuDNN的安装教程,参考文章。NVIDIA驱动程序CUDA+cuDNN安装教程
YOLOv4相关配置文件
链接:https://pan.baidu.com/s/1oh36WVqsBmr3_oC7JGBxeg
提取码:erx3
直接下载这么相关文件。
三、官方模型权重测试
以上环境全部搭建成功以后,就可以进行代码的测试了。在我们下载的YOLOv4相关配置文件中已经给出了测试要用到的代码opencv4_yolov4.cpp。我们新建一个C++项目
将主程序替换成opencv4_yolov4.cpp,并在程序所在位置新建两个文件夹m