[发明专利]一种信号灯的目标检测方法、装置、电子设备及存储介质

说明书

本发明公开了一种信号灯的目标检测方法、装置、电子设备及存储介质。该方法包括：获取待检测的目标图像；基于信号灯检测模型，确定目标图像中信号灯的类别和/或位置；其中，信号灯检测模型基于目标检测网络训练得到。本申请实施例的技术方案，可以解决现有技术中检测准确率及效率较低的问题，避免夜间、雨天、雾霾、有遮挡等困难场景对信号灯目标检测的影响，提高信号灯的目标检测效率及准确率，为信号灯的目标检测提供了一种新思路。

技术领域

本发明实施例涉及图像处理技术领域，尤其涉及一种信号灯的目标检测方法、装置、电子设备及存储介质。

背景技术

随着车辆闯红灯违法行为的不断增加，对交通卡口图像中的信号灯进行目标检测，成为智能交通系统中极其重要的环节。

目标检测即找出图像中所有感兴趣的物体，包含物体定位和物体分类两个子任务，同时确定物体的类别和位置。传统目标检测的方法一般分为三个阶段:首先在给定的图像上选择一些候选的区域，然后对这些区域提取特征，最后使用训练的分类器进行分类。现有一种滑动窗口方法进行目标位置的定位，即基于滑动窗口方法对交通卡口图像中的信号灯进行定位。因为目标可能出现在图像的任何位置，而且目标的大小、长宽比例也不确定，所以采用滑动窗口对整幅图像进行遍历，需要设置不同的尺度，不同的长宽比。这样，容易导致检测失败，降低检测成功率。同时，这种穷举的方法时间复杂度太高，遍历效率低，产生冗余窗口太多，严重影响后续特征提取及分类的速度和性能。

综上，传统的信号灯识别方法检测准确率及效率较低，无法在夜间、雨天、雾霾、有遮挡等困难场景下保证较高的检测准确率及效率，亟需改进。

发明内容

本发明提供一种信号灯的目标检测方法、装置、电子设备及存储介质，以提高信号灯的目标检测效率及准确率。

第一方面，本发明实施例提供了一种信号灯的目标检测方法，包括：

获取待检测的目标图像；

基于信号灯检测模型，确定所述目标图像中信号灯的类别和/或位置；

其中，所述信号灯检测模型基于目标检测网络训练得到。

第二方面，本发明实施例还提供了一种信号灯的目标检测装置，包括：

图像获取模块，用于获取待检测的目标图像；

类别位置确定模块，用于基于信号灯检测模型，确定所述目标图像中信号灯的类别和/或位置；

其中，所述信号灯检测模型基于目标检测网络训练得到。

第三方面，本发明实施例还提供了一种电子设备，该设备包括：

一个或多个处理器；

存储装置，用于存储一个或多个程序，

当所述一个或多个程序被所述一个或多个处理器执行，使得所述一个或多个处理器实现本发明任意实施例所述的信号灯的目标检测方法。

第四方面，本发明实施例还提供了一种计算机可读存储介质，其上存储有计算机程序，该程序被处理器执行时实现如本发明任意实施例所述的信号灯的目标检测方法。

本发明实施例提供的一种信号灯的目标检测方法、装置、电子设备及存储介质，通过获取待检测的目标图像；基于信号灯检测模型，确定所述目标图像中信号灯的类别和/或位置，其中，所述信号灯检测模型基于目标检测网络训练得到。本实施例的技术方案，可以解决现有技术中检测准确率及效率较低的问题，避免夜间、雨天、雾霾、有遮挡等困难场景对信号灯目标检测的影响，提高信号灯的目标检测效率及准确率，为信号灯的目标检测提供了一种新思路。

附图说明

图1为本发明实施例一提供的一种信号灯的目标检测流程图；