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实时检测
后续补充