[发明专利]一种信号灯重定位方法、装置、电子设备及存储介质

说明书

本申请公开了一种信号灯重定位方法，所述信号灯重定位方法包括：获取目标信号灯图像，并对所述目标信号灯图像进行语义分割得到前景图像；其中，所述目标信号灯图像为红灯点亮时的信号灯图像，所述前景图像为红灯所在区域的图像；对所述前景图像向预设方向进行投影，并根据投影结果确定所述前景图像中的信号灯位置参考点；根据所述信号灯位置参考点与信号灯配置框的当前参考点确定位置偏差，并根据所述位置偏差对所述信号灯配置框进行信号灯重定位。本申请能够提高重定位信号灯的精准度和效率。本申请还公开了一种信号灯重定位装置、一种电子设备及一种存储介质，具有以上有益效果。

技术领域

本申请涉及图像处理技术领域，特别涉及一种信号灯重定位方法、装置、一种电子设备及一种存储介质。

背景技术

由于交通信号灯是一个亮度较高的点光源，而电子警察等设备通常采用全局曝光的方式采集信号灯图像，但是上述图像采集方式可能会导致交通信号灯在成像的时候出现偏色情况，从而给执法部门在处理交通违章的过程中带来不必要的争议。为了解决该问题，行业上通常的做法都是采用信号灯增强的技术使得信号灯的成像在感官上更接近于真实效果，从而为违章判罚提供有力可靠的证据材料。

在红绿灯增强的技术当中，一般的做法是绘制信号灯配置框，进一步会在配置框的区域内通过图像处理等方法，最终实现对偏色红灯的效果增强。但是由于热胀冷缩等外界因素会导致信号灯在画面的位置发生偏移，需要对信号灯配置框的位置进行校正，才能保证红绿灯增强的最终效果。相关技术中，主要根据灯组颜色浓度及亮度结合信号灯周期性变换的特征计算信号灯中心点进而重定位信号灯配置框，这种方式具有可靠性差，耗时长等缺点。

因此，如何提高重定位信号灯的精准度和效率是本领域技术人员目前需要解决的技术问题。

发明内容

本申请的目的是提供一种信号灯重定位方法、装置、一种电子设备及一种存储介质，能够提高重定位信号灯的精准度和效率。

为解决上述技术问题，本申请提供一种信号灯重定位方法，该信号灯重定位方法包括：

获取目标信号灯图像，并对所述目标信号灯图像进行语义分割得到前景图像；其中，所述目标信号灯图像为红灯点亮时的信号灯图像，所述前景图像为红灯所在区域的图像；

对所述前景图像向预设方向进行投影，并根据投影结果确定所述前景图像中的信号灯位置参考点；

根据所述信号灯位置参考点与信号灯配置框的当前参考点确定位置偏差，并根据所述位置偏差对所述信号灯配置框进行信号灯重定位。

可选的，对所述前景图像向预设方向进行投影，并根据投影结果确定所述前景图像中的信号灯位置参考点，包括：

生成所述前景图像的二值化图像，并对所述二值化图像在水平方向和竖直方向进行投影，得到第一投影结果；其中，所述第一投影结果包括所述二值化图像在水平方向和竖直方向上每一位置的红灯图像像素点数量；

根据所述第一投影结果将红灯图像像素点数量小于第一预设值的像素点标记为噪点，并去除所述二值化图像中的噪点得到前景点感兴趣区域；其中，所述前景点感兴趣区域为所述二值化图像中红灯所在的区域；

对所述前景点感兴趣区域对应的二值化图像在水平方向和竖直方向进行投影，得到第二投影结果；其中，所述第二投影结果包括所述前景点感兴趣区域对应的二值化图像在水平方向和竖直方向上每一位置的红灯图像像素点数量；

根据所述第二投影结果将红灯图像像素点数量小于第二预设值的像素点标记为噪点，并去除所述前景点感兴趣区域中的噪点；

将所述前景点感兴趣区域中最左侧或最上方连通域的目标点设置为所述信号灯位置参考点。

可选的，将所述前景点感兴趣区域中最左侧或最上方连通域的目标点设置为所述信号灯位置参考点，包括：