『heqingchun-ubuntu编译使用darknet-yolov4』

ubuntu编译使用darknet-yolov4

一、更新、安装基础依赖

sudo apt update && \
sudo apt upgrade -y && \
sudo apt install -y build-essential

二、准备文件

1.下载文件

(1)darknet-master.zip

网址

https://github.com/AlexeyAB/darknet

点击"Code"下的"Download ZIP"

(2)yolov4.weights

网址

https://github.com/AlexeyAB/darknet

在网页找到yolov4.weights，并进行下载或者直接执行以下命令下载

wget https://github.com/AlexeyAB/darknet/releases/download/darknet_yolo_v3_optimal/yolov4.weights

2.新建目录

mkdir darknet

将(1)、(2)中下载的文件放入darknet目录
当前目录结构为
第一级：darknet
第二级：darknet-master.zip、yolov4.weights

三、解压复制文件

进入darknet目录

cd 你的具体路径/darknet

1.解压

unzip darknet-master.zip

2.复制文件

cp yolov4.weights darknet-master

四、make

1.make前修改Makefile文件

gedit Makefile

(1)

GPU=1
CUDNN=1
CUDNN_HALF=1
OPENCV=1
LIBSO=1

(2)

ARCH：将前边的"#"删掉，我的显卡是3050所以使用86这个，需要根据自己的实际去修改
查询网址：https://developer.nvidia.com/cuda-gpus#collapseOne

# GeForce RTX 3070, 3080, 3090
ARCH= -gencode arch=compute_86,code=[sm_86,compute_86]

(3)

NVCC

NVCC=/usr/local/cuda/bin/nvcc

五、编译

1.编译

make -j $(nproc)

六、验证

执行以下命令

./darknet detect build/darknet/x64/cfg/yolov4.cfg yolov4.weights build/darknet/x64/dog.jpg

弹出识别的图片即成功
darknet-yolov4编译完毕

七、Qt项目中使用yolov4实时检测

后续补充