yolov4的cfg文件怎么生成_YOLOv4 训练自己的数据集

1、克隆仓库下来。

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

2、对darknet文件夹下的Makefile进行修改。(主要是增加自己nvcc的路径，把CUDNN注释掉，不然容易报错，我不知道为啥)

修改好了之后，可以编译一下，编译成功，出现可执行文件darknet即可。

make # 编译

3、准备数据集：我这边的数据集是VOC数据集，需要把转换成YOLO格式的。在darknet文件夹下新建VOCdevkit文件夹。在VOCdevkit下新建VOC2007， VOC2007文件夹下新建Annotations(所有xml标注文件)、ImageSets(该文件夹下再建Main文件夹下，后期生生成train.txt等文件)、JPEGImages(所有图片)三个文件夹。

需要再Main文件夹下生成train.txt、valid.txt等，运行split_data.py，得到各个txt文件夹：

得到这些文件之后需要将xml标注文件转换成YOLO训练格式，运行voc_label.py。得到的标注文件是在labels文件夹下，并且会得到一些txt文件。

# split_data.py 将JPEGImages下的所有文件分成训练集，测试集，生成ImageSets下Main文件夹中的txt文件。

import os

import