一种基于自注意力机制的交通标志识别方法

本发明公开了一种基于自注意力机制的交通标志识别方法，包括如下步骤：S1：获取待识别交通标志的视频输入流，并输入至预先构建的基于自注意力机制的交通识别模型进行识别；S2:基于预先设置的种类的目标对视频输入流进行框选，得到交通标志的位置以及相对应的类别信息；其中，交通识别模型包括特征提取层、特征融合层和多尺度检测回归层；特征提取层依次连接的四层，用于对视频输入流的每一帧图像进行分割、展平、映射、编码操作，并提取特征图；特征融合层用于对特征提取层进行上采样，对特征图进行特征融合并划分出不同的尺度；多尺度检测回归层用于对不同尺度的特征图进行检测和种类识别。本发明的自注意力机制鲁棒性更好、精度更高。
上海应用技术大学
周峰 | 刘云翔 | 原鑫鑫