YOLO v4论文:https://arxiv.org/abs/2004.10934
YOLO v4源码:https://github.com/AlexeyAB/darknet
给学妹写的教程,发上来作为学习记录吧,数据集用的是开源车牌数据集CCPD,本文章介绍的只是检测车牌框,并不带车牌识别功能。有错误或者有疑问的欢迎底下评论。
文章目录
1、环境配置(亲测可用)
- Ubuntu 20.04
- CUDA 11.2
- cuDNN
- OpenCV_contrib 4.3.0
- OpenCV 4.3.0
2、下载安装
1.CUDA 11.2
a.禁用集成的nouveau驱动
gedit打开blacklist.conf
sudo gedit /etc/modprobe.d/blacklist.conf
blacklist.conf末尾添加:
blacklist nouveau
options nouveau modeset=0
更新系统
sudo update-initramfs -u
重启系统
reboot
验证nouveau是否禁用(没有任何输出说明已经禁用)
lsmod | grep nouveau
b.运行cuda文件
sudo sh cuda_11.2.0_460.27.04_linux.run
如果之前已经安装过显卡驱动,这里就不需要选择第一个了
Install
添加环境变量
sudo gedit ~/.bashrc
在最后一行加上
export PATH=/usr/local/cuda-11.2/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-11.2/lib64:$LD_LIBRARY_PATH
激活环境变量
source ~/.bashrc
检测是否安装成功
nvidia-smi
nvcc -V
- cuDNN下载链接:官网下载(需要登录)
百度网盘链接 提取码:80rv - 解压cuDNN压缩包,并在文件夹下的cuda文件夹下打开终端
将include和lib64文件夹下的内容复制到CUDA安装目录下对应名称的文件夹下
sudo cp -r include/* /usr/local/cuda/include/
sudo cp -r lib64/* /usr/local/cuda/lib64/
2.OpenCV
$ sudo apt-get install -y build-essential checkinstall cmake pkg-config yasm git gfortran
# 添加源,防止可能会出现的找不到所需安装的库的情况
$ sudo add-apt-repository -y "deb http://security.ubuntu.com/ubuntu xenial-security main"
$ sudo apt-get update
$ sudo apt-get install -y libgstreamer1.0-dev libgstreamer-plugins-base1.0-dev
$ sudo apt-get install -y libjpeg8-dev libjasper-dev libpng12-dev libtiff5-dev libavcodec-dev libavformat-dev libswscale-dev libdc1394-22-dev libxine2-dev libv4l-dev
$ sudo apt-get install -y libgstreamer1.0-dev libgstreamer-plugins-</