车辆信号灯的识别方法、识别模型的训练方法及相关设备技术

本申请公开了一种车辆信号灯的识别方法、识别模型的训练方法及相关设备。该方法包括：获取车辆的目标图像；对目标图像进行目标部位的信号灯的检测，得到至少一个目标部位信号灯区域；利用识别模型对至少一个目标部位信号灯区域进行处理，获取至少一个对应的目标部位信号灯的状态信息，其中，识别模型是利用至少两个不同类型的目标部位信号灯区域训练得到的。上述方案，能够提高目标部位信号灯的状态信息识别的准确度。识别的准确度。识别的准确度。

Recognition method of vehicle signal lamp, training method of recognition model and related equipment

【技术实现步骤摘要】
车辆信号灯的识别方法、识别模型的训练方法及相关设备


[0001]本申请涉及图像处理
，特别是涉及一种车辆信号灯的识别方法、车辆信号灯的识别模型的训练方法、电子设备及存储装置。

技术介绍

[0002]随着社会经济的发展和国民生活水平的提高，全国机动车保有量逐年递增，在城市道路中的车辆安全行驶也愈发重要。道路上的汽车通过信号灯或车灯发出行驶信号，以对其他车辆进行信号提醒。
[0003]目前，由于车辆的信号灯受环境影响，例如强烈光照、信号灯遮挡等问题，无法准确地对车辆的信号灯进行识别，从而影响车辆的行驶安全，容易引发交通事故等问题。

技术实现思路

[0004]本申请主要解决的技术问题是提供一种车辆信号灯的识别方法、识别模型的训练方法及相关设备，能够提高目标部位信号灯的状态信息识别的准确度。
[0005]为了解决上述问题，本申请第一方面提供了一种车辆信号灯的识别方法，该方法包括：获取车辆的目标图像；对目标图像进行目标部位的信号灯的检测，得到至少一个目标部位信号灯区域；利用识别模型对至少一个目标部位信号灯区域进行处理本文档来自技高网...

【技术保护点】

【技术特征摘要】
1.一种车辆信号灯的识别方法，其特征在于，所述方法包括：获取车辆的目标图像；对所述目标图像进行目标部位的信号灯的检测，得到至少一个目标部位信号灯区域；利用识别模型对所述至少一个目标部位信号灯区域进行处理，获取至少一个对应的目标部位信号灯的状态信息，其中，所述识别模型是利用至少两个不同类型的目标部位信号灯区域训练得到的。2.根据权利要求1所述的方法，其特征在于，所述至少一个目标部位信号灯区域包括不同类型的信号灯对应的N个部位信号灯区域，所述N为大于1的整数；所述利用识别模型对所述至少一个目标部位信号灯区域进行处理，获取至少一个对应的目标部位信号灯的状态信息，包括：利用所述识别模型中的第一预设网络，分别对所述N个部位信号灯区域进行特征提取，得到所述N个部位信号灯区域的部位特征；对所述N个部位信号灯区域的部位特征进行重组处理，得到重组特征；利用所述识别模型中的第二预设网络对所述重组特征进行处理，得到对应的部位信号灯的状态信息。3.根据权利要求2所述的方法，其特征在于，所述对所述N个部位信号灯区域的部位特征进行重组处理，得到重组特征，包括：对各个所述部位特征分别进行依序进行卷积处理、注意力处理，得到各个部位特征对应的卷积特征；基于所述各个部位特征对应的卷积特征，得到表征卷积特征；利用所述表征卷积特征和所述各个部位特征中的至少部分部位特征进行注意力处理，得到所述重组特征。4.根据权利要求3所述的方法，其特征在于，所述N为2，所述N个部位信号灯区域包括第一部位信号灯区域和第二部位信号灯区域；所述利用所述表征卷积特征和所述各个部位特征中的至少部分部位特征进行注意力处理，得到所述重组特征，包括：利用所述表征卷积特征和第一部位特征进行注意力处理，得到所述重组特征；其中，所述第一部位特征是利用所述第一预设网络对所述第一部位信号灯区域进行特征提取得到的。5.根据权利要求1所述的方法，其特征在于，当所述至少一个目标部位信号灯区域仅包括一个类型的部位信号灯区域时，将所述一个类型的部位信号灯区域作为待处理部位信号灯区域；所述利用识别模型对所述至少一个目标部位信号灯区域进行处理，获取至少一个对应的目标部位信号灯的状态信息，包括：利用所述识别模型中的第一预设网络对所述待处理部位信号灯区域进行特征提取，得到所述待处理部位信号灯区域的待处理部位特征；利用分类层对所述待处理部位特征进行处理，得到所述待处理部位信号灯的状态信息；或者，利用所述识别模型的第三预设网络对所述待处理部位信号灯区域进行处理，得到对应的所述待处理部位信号灯的状态信息...

【专利技术属性】
技术研发人员：何正海，林骏，王亚运，
申请(专利权)人：浙江大华技术股份有限公司，
类型：发明
国别省市：