YOLOv4_车牌号检测.zip

时间:2024-09-26 03:21:59
【文件属性】:

文件名称:YOLOv4_车牌号检测.zip

文件大小:725.76MB

文件格式:ZIP

更新时间:2024-09-26 03:21:59

1、YOLOv4 2、车牌号检测

以下4个py文件需要单独运行: 首先运行annotation.py文件:将CCPD数据集进行划分,生成.txt路径文件存储到model_data文件夹中。 然后运行k_means_calculate.py文件:计算生成anchors数值,存储到model_data文件夹中。 再然后运行train.py文件:加载原始权重,训练YOLOv4模型,并将每轮训练的结果存储进Logs文件夹中。 最后运行yolo_predict.py文件:载入训练好的YOLOv4权重,对测试集数据进行检测,检测结果存放入demo文件夹中。 算法效果:---------------------------------------------------------------------- YOLOv4每张图片检测耗时323.8ms,精度较高,训练40 epoch左右后val loss降低至0.3附近,效果非常满意。 训练过程中random数据增强非常重要,可大幅提升模型性能,进一步降低val loss。


网友评论