摘要
本发明公开了一种基于多尺度特征融合的交通标志检测方法,包括S1:构建数据集,数据集包括若干交通标志图片及对应的标签;S2:构建检测模型,导入S1中数据集的图片至色差特征提取模块中,获取色差特征;S3:导入S2中的色差特征至骨干特征提取中,获取不同尺度的提取特征;S4:导入S3中的提取特征至多视图协同融合模块中,获取多视图协助融合特征;S5:将多视图协同融合特征和色差特征进行残差连接后输入至检测头中,获取预测输出;S6:优化模型参数。本发明通过对色差特征和多视图协同融合特征进行残差连接,从而结合交通标志图片的色彩信息和视觉信息在识别过程中进行了图像补全,从而能获取更准确的检测结果。
技术关键词
交通标志检测方法
多尺度特征融合
融合特征
特征提取模块
多视角特征
压缩特征
图片
注意力
积层
蓝色
通道
分量色差
红色
数据
矩阵
语义