1.下载yolov4
git clone https://github.com/AlexeyAB/darknet.git
如果没有git
sudo apt-get install git
2.编译
进入darknet的目录下
执行下面的语句进行编译
make 或者 make -j8
3.下载yolov4权重文件
https://drive.google.com/open?id=1cewMfusmPjYWbrnuJRuKhPMwRe_b9PaT
4.运行样例
./darknet detect cfg/yolov4.cfg yolov4.weights data/dog.jpg
此时已经可以预测照片了,但是会有OpenCV为编译的现象,如果想编译OpenCV的话就接着往下看
5.修改Makefile
6.下载一些第三方库
安装opencv
pip install opencv-python -i https://pypi.tuna.tsinghua.edu.cn/simple
安装opencv依赖库
sudo apt-get update
sudo apt-get install libopencv-dev
7.修改nvcc文件路径(在Makefile中)
修改为本机的cuda路径
8.重新编译
进入darknet的目录下
执行下面的语句进行编译
make 或者 make -j8
9.运行
./darknet detect cfg/yolov4.cfg yolov4.weights data/dog.jpg
如果是在xshell这种远程命令行连接工具中使用上面的代码,会提示 Unable to init server: Could not connect: Connection refused,这是由于没有可视化界面,弹窗无法显示,在可视化界面中的命令行中运行就正常了