计算机系统应用教程网站

网站首页 > 技术文章 正文

Ubuntu配置darknet实现YOLOv4目标检测(一)——配置darknet

btikc 2024-09-12 12:01:56 技术文章 21 ℃ 0 评论

目录

1. 概述

2.darknet配置

2.1 下载darknet

2.2 编译darknet

3.测试

3.1 测试图片


1. 概述

YOLO官方网址:https://pjreddie.com/darknet/yolo/

YOLOv1~v3作者开源项目地址:https://github.com/pjreddie/darknet

YOLOv4作者开源项目地址:https://github.com/AlexeyAB/darknet

YOLOv1论文地址:https://arxiv.org/pdf/1506.02640.pdf

YOLOv2论文地址:https://arxiv.org/pdf/1612.08242.pdf

YOLOv3论文地址:https://arxiv.org/pdf/1804.02767.pdf

YOLOv4论文地址:https://arxiv.org/pdf/2004.10934.pdf

2.darknet配置

2.1 下载darknet

在Ubuntu终端输入以下命令:

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

2.2 编译darknet

在编译darknet前首先需要修改Makefile,在终端输入以下命令:

cd darknet
gedit Makefile

打开后根据需要修改GPU,CUDNN等的值,GPU为1的话需要安装cuda,CUDNN为1的话需要安装CUDNN,CUDNN_HALF仅支持带有Tensor Cores的GPU如Titan V / Tesla V100 / DGX-2或者之后的显卡。

OPENCV为1的话,需要安装OpenCV,安装OpenCV可以在终端中输入以下命令:

sudo apt-get install libopencv-dev

通过编译安装opencv的话可以参考这篇博客的安装步骤:

ubuntu 18.04编译安装opencv 4.3.0_令狐傻笑的博客-CSDN博客

修改完Makefile之后,在终端中输入:

make

或者

make -j8

进行编译,j后面的数字表示cpu核数,cpu有多少个核就可以写几。

3.测试

3.1 测试图片

编译成功之后,进行测试,首先需要下载权重,权重下载地址:

https://drive.google.com/open?id=1cewMfusmPjYWbrnuJRuKhPMwRe_b9PaT

完成权重下载后,可以写一个shell的脚本文件,在终端中输入

touch test_coco.sh
gedit test_coco.sh

在test_coco.sh文件中写入以下内容:

./darknet detector test cfg/coco.data cfg/yolov4.cfg yolov4.weights -thresh 0.25 ./data/dog.jpg

在终端中输入以下命令后可以看到:

./test_coco.sh

完成检测后,可以得到如下结果,说明配置成功。

终端输出的结果如下,输出了每种类别的置信度。

data/dog.jpg: Predicted in 21.041000 milli-seconds.
bicycle: 92%
dog: 98%
truck: 92%
pottedplant: 33%

下图是YOLOv3的结果,YOLOv4比YOLOv3多检测了一个pottedplant。

终端输出的结果如下

data/dog.jpg: Predicted in 17.278000 milli-seconds.
bicycle: 100%
dog: 100%
truck: 94%


?

本文暂时没有评论,来添加一个吧(●'◡'●)

欢迎 发表评论:

最近发表
标签列表