[发明专利]交通信号灯灯色、颜色识别方法、电子设备及存储介质

说明书

本发明涉及图像处理技术领域，具体涉及交通信号灯灯色、颜色识别方法、电子设备及存储介质，其中灯色识别方法包括获取目标交通信号灯的待检测灯色位置以及已检测灯色位置；利用待检测灯色位置与已检测灯色位置的关系，对已检测灯色位置进行更新，以确定目标交通信号灯的各个目标灯色位置；基于目标交通信号灯的各个目标灯色位置，确定各个目标灯色位置对应的灯色。利用待检测灯色位置与已检测灯色位置的关系对已检测灯色位置进行更新，可以较准确的得到目标交通信号灯的各个目标灯色位置及其对应的灯色；后续基于目标交通信号灯的各个目标灯色位置及其对应的灯色就可以准确地识别出目标交通信号灯的颜色，提高了交通信号灯颜色识别的准确率。

技术领域

本发明涉及图像处理技术领域，具体涉及交通信号灯灯色、颜色识别方法、电子设备及存储介质。

背景技术

在一些智能控制领域需要对交通信号灯的颜色进行识别，从而实现对交通设备的智能控制。例如，在无人驾驶领域，需要基于识别出的交通信号灯的颜色，对交通设备进行相应的控制，以保证交通设备的安全行驶。

现有技术中一般通过采集交通路口的交通信号灯的图像，利用深度学习方法识别检测图像中交通信号灯的颜色。例如，获取交通信号灯的检测图像，将检测图像输入检测网络中，输出检测图像对应的交通信号灯的颜色。然而，这一识别方法中，由于基于可见光的摄像头在采集交通路口的车流状况时，因为过曝或反光灯原因，导致交通信号灯检测算法误判，从而导致交通信号灯的颜色识别错误。

发明内容

有鉴于此，本发明实施例提供了一种交通信号灯灯色、颜色识别方法、电子设备及存储介质，以解决交通信号灯颜色识别的准确率偏低的问题。

根据第一方面，本发明实施例提供了一种交通信号灯灯色识别方法，包括：

获取目标交通信号灯的待检测灯色位置以及已检测灯色位置；

利用所述待检测灯色位置与所述已检测灯色位置的关系，对所述已检测灯色位置进行更新，以确定所述目标交通信号灯的各个目标灯色位置；

基于所述目标交通信号灯的各个目标灯色位置，确定各个所述目标灯色位置对应的灯色。

本发明实施例提供的交通信号灯灯色识别方法，由于交通信号灯中各个灯色之间的相对位置以及各个位置的灯色是固定的且不会随着外界环境变化而变化，那么利用待检测灯色位置与已检测灯色位置的关系对已检测灯色位置进行更新，可以较准确的得到目标交通信号灯的各个目标灯色位置，在目标灯色位置确定之后，就可以较准确地确定各个目标灯色位置对应的灯色；后续基于目标交通信号灯的各个目标灯色位置及其对应的灯色就可以准确地识别出目标交通信号灯的颜色，提高了交通信号灯颜色识别的准确率。

结合第一方面，在第一方面第一实施方式中，所述利用所述待检测灯色位置与所述灯色位置的关系，对所述已检测灯色位置进行更新，以确定所述目标交通信号灯的目标灯色位置，包括：

判断所述待检测灯色位置与各个所述已检测灯色位置的关系是否满足预设条件；

当所述待检测灯色位置与所述已检测灯色位置的关系满足所述预设条件时，将所述已检测灯色位置更新至所述待检测灯色位置，以确定所述目标交通信号灯的目标灯色位置。

本发明实施例提供的交通信号灯灯色识别方法，在待检测灯色位置与已检测灯色位置的关系满足预设条件时，实时对已检测灯色位置进行更新，可以提高确定出的目标灯色位置的准确性。

结合第一方面第一实施方式，在第一方面第二实施方式中，所述已检测灯色位置包括已检测灯色在所述目标交通信号灯中的所属区域以及所述已检测灯色在所述所属区域中的位置；其中，所述判断所述待检测灯色位置与各个所述已检测灯色位置的关系是否满足预设条件，包括：

利用所述待检测灯色位置，确定待检测灯色在所述交通信号灯中的所属区域；