一种用于提升极端天气与环境下交通标志识别精度的方法

摘要

本发明公开了一种提升极端天气与环境下交通标志识别精度的方法，以YoloV5目标检测模型为基础，融入聚焦模块、跨阶段局部融合模块和空间金字塔池化结构，对于光线不好的交通标志图像，能够更好地从局部特征来提取特征图信息，特征图更精准地表达了图像。对于数量不多的训练数据，本发明使用高斯噪声、添加椒盐噪声、减少亮度、锐化图像、尺寸大小等比例缩小来模拟交通标志在不同环境下的表现，并使用复制‑粘贴的方法将交通标志复制到无目标的图片上，极大地丰富了数据集。使用本发明的方法更容易地捕捉不同分辨率下的不同图像模式，能够使得目标的特征得到最大程度的提取和融合；同时使得收敛更快速和准确，有更少的定位错误，产生更精确的预测。