基于多尺度像素特征融合的实时小交通标志检测算法

摘要

交通标志检测技术是先进驾驶辅助系统中重要组成部分.真实的驾驶环境中要求交通标志检测系统具备极高的实时性与准确性.轻量级网络MobileNetv2-SSD能够满足检测的实时性,但准确性不足以满足实际需求.本文将MobileNetv2-SSD作为基础网络,提出一种基于像素重排的多尺度像素特征融合方法,并在网络的检测层引入高效通道注意力机制,实现特征增强.在保证算法的实时性的同时,有效提升了小交通标志的检测性能.实验结果表明,本文算法模型能够在真实环境下准确实时地检测小交通标志.在长沙理工大学中国交通标志检测数据集CCTSDB上取得93.2％的mAP,模型大小仅为17.3M,检测每张图像的时间为0.022 s.