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

权利要求书

1.一种交通信号灯的检测方法，其特征在于，包括：

采集交通信号灯图像；

在所述交通信号灯图像上对交通信号灯位置和交通信号灯类别进行标注，得到交通信号灯标注图像与交通信号灯标注文件，其中，所述交通信号灯标注文件用于存储所述交通信号灯标注图像中的标注真值；

对所述交通信号灯标注图像进行过采样处理得到交通信号灯预处理图像；

从所述交通信号灯标注文件中确定交通信号灯位置区域；

将所述交通信号灯位置区域复制到所述交通信号灯预处理图像中的贴图区域，得到交通信号灯中间图像，其中，所述贴图区域与所述交通信号灯预处理图像中已有的交通信号灯目标不存在重叠区域；

对所述交通信号灯中间图像进行检测，得到目标交通信号灯检测结果。

2.根据权利要求1所述的检测方法，其特征在于，对所述交通信号灯中间图像进行检测，得到目标交通信号灯检测结果，具体包括：

对所述交通信号灯中间图像进行图像增强处理，得到目标交通信号灯图像；

对所述目标交通信号灯图像进行检测，得到所述目标交通信号灯检测结果。

3.根据权利要求2所述的检测方法，其特征在于，对所述目标交通信号灯图像进行检测，得到所述目标交通信号灯检测结果，具体包括：

采用YOLOV3检测方法从所述目标交通信号灯图像中检测出交通信号灯，记为交通信号灯检测的第一次输出结果；

采用roi增强方式从所述目标交通信号灯图像中检测出交通信号灯，记为交通信号灯检测的第二次输出结果；

将所述第一次输出结果和第一次输出结果进行融合，得到所述目标交通信号灯检测结果。

4.根据权利要求3所述的检测方法，其特征在于，所述YOLOV3检测方法采用CNN网络，所述CNN网络中的神经网络模型主体结构Backbone采用轻量化结构Xception。

5.一种交通信号灯的检测装置，其特征在于，包括：

采集单元，用于采集交通信号灯图像；

标注单元，用于在所述交通信号灯图像上对交通信号灯位置和交通信号灯类别进行标注，得到交通信号灯标注图像与交通信号灯标注文件，其中，所述交通信号灯标注文件用于存储所述交通信号灯标注图像中的标注真值；

过采样处理单元，用于对所述交通信号灯标注图像进行过采样处理得到交通信号灯预处理图像；

区域确定单元，用于从所述交通信号灯标注文件中确定交通信号灯位置区域；

区域复制单元，用于将所述交通信号灯位置区域复制到所述交通信号灯预处理图像中的贴图区域，得到交通信号灯中间图像，其中，所述贴图区域与所述交通信号灯预处理图像中已有的交通信号灯目标不存在重叠区域；

检测单元，用于对所述交通信号灯中间图像进行检测，得到目标交通信号灯检测结果。

6.根据权利要求5所述的检测装置，其特征在于，所述检测单元具体包括：

增强处理子单元，用于对所述交通信号灯中间图像进行图像增强处理，得到目标交通信号灯图像；

检测子单元，用于对所述目标交通信号灯图像进行检测，得到所述目标交通信号灯检测结果。

7.根据权利要求6所述的检测装置，其特征在于，所述检测子单元具体用于：

采用YOLOV3检测方法从所述目标交通信号灯图像中检测出交通信号灯，记为交通信号灯检测的第一次输出结果；

采用roi增强方式从所述目标交通信号灯图像中检测出交通信号灯，记为交通信号灯检测的第二次输出结果；

将所述第一次输出结果和第一次输出结果进行融合，得到所述目标交通信号灯检测结果。

8.根据权利要求7所述的检测装置，其特征在于，所述YOLOV3检测方法采用CNN网络，所述CNN网络中的神经网络模型主体结构Backbone采用轻量化结构Xception。