一种基于YOLOv4算法的交通标志检测方法

    公开(公告)号:CN113408410A

    公开(公告)日:2021-09-17

    申请号:CN202110676065.8

    申请日:2021-06-18

    Abstract: 本发明公开了一种基于YOLOv4算法的交通标志检测方法,涉及图像技术领域,所述方法包括步骤:通过原始YOLOv4算法对预处理后的交通标志图像进行模型训练;通过训练迭代,把模型训练参数达到最优且将损失函数达到最小的模型保存;通过训练好的模型对测试集中的图像进行测试,最终选择置信度最高的边框进行输出,完成交通标志的检测,本发明中改进后的YOLOv4算法能够达到对远处小型交通标志的检测准确率,并且改进后的YOLOv4模型的检测速度提升近3倍,并且极大的减少了模型的参数量。

Patent Agency Ranking