这两天尝试编译了一下新出炉的YOLOV4,下一步继续尝试训练自己的数据。
源码下载编译过程主要参考了这篇博客
一、安装CUDA,cudnn,OPENCV等
基本环境要求:cuda=10.0,cudnn>=7.0, opencv>=2.4
参考了自己之前安装CUDA,cudnn,OPENCV的博客以及一些别人的方法。
最终我的电脑配置如下:RTX2080ti,CUDA10.2,cudnn7.6.5
在这期间,报错出现的包都要自行补上。
如果需要升级cmake升级参考了这篇博客
二、下载源码
阅读论文:YOLO论文
github源代码:GitHub - AlexeyAB/darknet: YOLOv4
1.下载后解压,编译
cd darknet-master
mkdir build-release
cd build-release
cmake ..
make
make install
2.单张测试
(1)首先需要下载yolov4的权重文件,放在/darknet文件夹中即可
(2)修改cfg里的配置文件
(3)编译后会生成一个darknet的可执行文件,在相应目录运行
./darknet detect cfg/yolov4.cfg yolov4.weights data/horses.jpg
检测结果后显示在屏幕上,并且保存在predicted.jpg可以独立查看
参考博客:
1.Ubuntu18.04下复现YOLOv4(YOLOv3亦可)
2.基于Ubuntu的yolov4 环境配置以及效果实测
3.linux下配置运行yolov4!亲测有效!