2303_78263643的博客

原创 使用yolov4模型训练数据集

cfg数据集的准备数据集的处理修改voc_annotation.py里面的annotation_mode=2，运行voc_annotation.py生成根目录下的2007_train.txt和2007_val.txt。开始网络训练train.py的默认参数用于训练VOC数据集，直接运行train.py即可开始训练。训练结果预测训练结果预测需要用到两个文件，分别是yolo.py和predict.py。