识别交通标志的方法、装置和电子设备与流程

本申请涉及人工智能，尤其涉及一种识别交通标志的方法、装置和电子设备。

背景技术：

1、随着计算机技术和人工智能技术的快速发展，人工智能技术被应用到越来越多的场景中，如智慧交通、图像识别等。

2、为了实现自动驾驶，准确识别交通标志具有重要作用。例如，自动驾驶车辆可以根据交通法规，按照交通标志进行并道、转弯、限速等操作。相关技术可以通过图像识别得到交通标志。

3、申请人发现，相关技术对某些特殊场景下的交通标志的识别效果有待提升。例如，拍摄图像中的交通标志图像被部分遮挡，或者由于路况等原因导致拍摄图像中的交通标志图像的至少部分区域不清晰或发生形变等，造成交通标志识别结果的准确度无法满足用户需求。

技术实现思路

1、为解决或部分解决相关技术中存在的问题，本申请提供一种识别交通标志的方法、装置和电子设备，能够有效提升交通标志图像部分缺失或者交通标志图像中部分图像质量较差场景下的识别效果。

2、本申请的第一个方面提供了一种识别交通标志的方法，包括：获得待识别图像；利用经训练的交通标志识别模型处理待识别图像，得到交通标志；其中，交通标志识别模型包括：特征图提取模块，用于从待识别图像中提取特征图，以及处理待识别图像和/或至少部分特征图，得到部分信息缺失的调整特征图；融合分类模块，用于对特征图和调整特征图进行融合，并基于融合后的特征图和调整特征图确定交通标志。

3、根据本申请的某些实施例，特征图提取模块包括：卷积神经网络，包括依次串联的输入层和至少两个卷积层，用于对待识别图像进行卷积操作，得到特征图；图处理单元，包括多个处理子单元，分别与输入层或者卷积层相连，用于处理待识别图像和/或至少部分卷积层输出的特征图，得到部分信息缺失的待识别图像和/或特征图。

4、根据本申请的某些实施例，融合分类模块包括：第一特征融合单元，用于融合部分信息缺失的待识别图像和/或特征图，得到调整特征图；第二特征融合单元，用于拼接特征图和调整特征图，以基于拼接后的特征图和调整特征图确定交通标志。

5、根据本申请的某些实施例，多个处理子单元各自对应有卷积核，卷积核的至少一个元素值为零。

6、根据本申请的某些实施例，卷积核的除零之外的元素符合高斯分布。

7、根据本申请的某些实施例，图处理模块还包括：插值单元，用于对部分信息缺失的特征图进行双线性插值，得到尺寸相同的待识别图像和/或特征图；特征融合单元具体用于融合尺寸相同的待识别图像和/或特征图，得到调整特征图。

8、根据本申请的某些实施例，上述方法还包括：关联交通标志图像数据和标注数据，生成样本数据；对样本数据进行随机分组，得到训练数据和测试数据；利用训练数据训练交通标志识别模型后，利用经训练的交通标志识别模型处理测试数据，得到测试结果；比对测试结果和测试数据的标签数据，确定交通标志识别模型输出识别结果的精准度。

9、本申请的第二方面提供了一种识别交通标志的装置，包括：图像获得模块和图像识别模块。其中，图像获得模块用于获得待识别图像；图像识别模块，用于利用经训练的交通标志识别模型处理待识别图像，得到交通标志；其中，交通标志识别模型包括：特征图提取模块，用于从待识别图像中提取特征图，以及处理待识别图像和/或至少部分特征图，得到部分信息缺失的调整特征图；融合分类模块，用于对特征图和调整特征图进行融合，并基于融合后的特征图和调整特征图确定交通标志。

10、本申请的第三方面提供了一种电子设备，包括：处理器；存储器，其上存储有可执行代码，当上述可执行代码被处理器执行时，使得处理器执行上述方法。

11、本申请的第四方面还提供了一种计算机可读存储介质，其上存储有可执行代码，当可执行代码被电子设备的处理器执行时，使处理器执行上述方法。

12、本申请的第五方面还提供了一种计算机程序产品，包括可执行代码，可执行代码被处理器执行时实现上述方法。

13、本申请提供的识别交通标志的方法、装置和电子设备，通过提取待识别图像中目标对象的特征来识别交通标志。其中，在提取目标对象的特征时，考虑了因遮挡、路况等原因导致的交通标志图残缺或部分图像质量差等问题。交通标志识别模型在提取图像特征时主动丢弃待识别图像的部分信息和/或特征图的部分信息，模拟了交通标志图残缺或质量差等场景。通过上述方式使得经训练的交通标志识别模型提取的调整特征图能够更好地应对交通标志残缺或者部分图像质量差的场景。本申请实施例能够有效提升交通标志图像部分缺失或者交通标志图像中部分图像质量较差场景下的识别效果。

14、应当理解的是，以上的一般描述和后文的细节描述仅是示例性和解释性的，并不能限制本申请。

技术特征：

1.一种识别交通标志的方法，其特征在于，包括：

2.根据权利要求1所述的方法，其特征在于，所述特征图提取模块包括：

3.根据权利要求2所述的方法，其特征在于，所述融合分类模块包括：

4.根据权利要求2所述的方法，其特征在于，多个所述处理子单元各自对应有卷积核，至少部分所述卷积核的至少一个元素值为零。

5.根据权利要求4所述的方法，其特征在于，所述卷积核的除零之外的元素符合高斯分布。

6.根据权利要求3所述的方法，其特征在于，所述图处理单元还包括：

7.根据权利要求1至6任一项所述的方法，其特征在于，还包括：

8.一种识别交通标志的装置，其特征在于，包括：

9.一种电子设备，其特征在于，包括：

10.一种计算机存储介质，其特征在于，存储有可执行代码，当所述可执行代码被执行时，执行根据权利要求1-8中任一项所述的方法。

技术总结
本申请涉及了一种识别交通标志的方法、装置和电子设备。该识别交通标志的方法包括：获得待识别图像；利用经训练的交通标志识别模型处理待识别图像，得到交通标志；其中，交通标志识别模型包括：特征图提取模块，用于从待识别图像中提取特征图，以及处理待识别图像和/或至少部分特征图，得到部分信息缺失的调整特征图；融合分类模块，用于对特征图和调整特征图进行融合，并基于融合后的特征图和调整特征图确定交通标志。本申请能够提升交通标志图像残缺、部分交通标志的图像质量差的交通标志的识别效果。

技术研发人员：李宁,万如,贾双成,郭杏荣
受保护的技术使用者：智道网联科技（北京）有限公司
技术研发日：
技术公布日：2024/1/13