yolov4训练自己的数据集

前几天(目前2020-04-30)yolov4轰轰烈烈的问世了,自己跑了一下顺便记录下来。
论文
代码

环境

Ubuntu 16.04
Python3.5
cuda 9

首先把代码下载下来，可以用下面命令下载，也可以直接点击上面代码链接下载。

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

•  

1.编译

如果需要使用GPU加速，那么得打开项目里面的makefile文件修改一些参数的值。修改完成之后在直接make。
GPU=1
CUDNN=1
CUDNN_HALF=1
OPENCV=1
OPENMP=1
LIBSO=1
DEBUG=1

# cd到darknet-master目录下
make
或者 make -j8

2.测试一下开源权重

# 测试图片，结果保存在darknet-master/predictions.jpg
./darknet detect cfg/yolov4.cfg