交通信号灯识别方法、装置、存储介质及电子设备与流程

本申请涉及一种图像处理，尤其涉及一种交通信号灯识别方法、装置、存储介质及电子设备。

背景技术：

1、在车辆行经有交通信号灯的路口时,需要识别交通信号灯的颜色，进而根据交通信号灯颜色的指示进行相应驾驶操作。

2、目前识别交通信号灯的颜色主要依赖于对交通信号灯的颜色识别，然而由于逆光的光线变化影响，对交通信号灯的颜色识别往往存在识别准确率较低的问题。因此在逆光场景下，如何识别交通信号灯，一直是本领域技术人员急切解决的问题。

技术实现思路

1、有鉴于此，有必要提供一种交通信号灯识别方法、装置、存储介质及电子设备，在逆光场景下，可以识别交通信号灯，也可提升交通信号灯识别的准确率。

2、第一方面本申请提供一种交通信号灯识别方法，所述方法包括：

3、获取包括交通信号灯的第一图像；

4、检测到所述第一图像处于逆光场景，将所述第一图像中的第一区域分割出来得到分割图，其中所述第一区域为一组交通信号灯图像区域；

5、提取所述分割图中点亮区域；

6、根据所述点亮区域在所述分割图中的位置来识别所述一组交通信号灯的点亮颜色。

7、在一些实施例中，所述提取所述分割图中点亮区域包括：

8、将所述分割图中的像素的rgb色域转换为hsv色域，得到转换后的分割图；

9、使用预设阈值提取所述转换后的分割图中的所述点亮区域。

10、在一些实施例中，所述使用预设阈值提取所述转换后的分割图中的所述点亮区域包括：

11、针对所述转换后的分割图的中每一像素，获取所述像素的v通道的数值；

12、将所述像素的v通道的数值大于所述预设阈值的像素作为第一像素；

13、筛选出第一像素所在的区域得到点亮区域。

14、在一些实施例中，所述方法还包括：

15、获取所述转换后的分割图的中所有像素的v通道的数值，得到总数值；

16、对所述总数值求平均得到预设阈值。

17、在一些实施例中，所述根据所述点亮区域在所述分割图中的位置来识别所述一组交通信号灯的点亮颜色包括：

18、将所述分割图均等划分为三个区域，其中所述三个区域分别对应所述一组交通信号灯中的三个信号灯；

19、获取所述点亮区域的位置和所述三个区域的位置；

20、根据所述点亮区域的位置和所述三个区域的位置，将所述三个区域中所述点亮区域位于的区域作为目标区域；

21、输出所述目标区域对应的颜色。

22、在一些实施例中，所述检测到所述第一图像处于逆光场景，将所述第一图像中的第一区域分割出来得到分割图包括：

23、检测到所述第一图像处于逆光场景，使用分割网络将所述第一图像中的第一区域分割出来得到分割图。

24、在一些实施例中，所述检测到所述第一图像处于逆光场景包括：

25、对所述第一图像中的所述交通信号灯进行颜色识别；

26、当无法识别所述交通信号灯的颜色时，确定所述第一图像处于逆光场景。

27、第二方面，本申请实施例提供一种交通信号灯识别装置，所述装置包括：

28、图像获取模块，用于获取包括交通信号灯的第一图像；

29、分割模块，用于检测到所述第一图像处于逆光场景，将所述第一图像中的第一区域分割出来得到分割图，其中所述第一区域为一组交通信号灯图像区域；

30、提取模块，用于提取所述分割图中点亮区域；

31、识别模块，用于根据所述点亮区域在所述分割图中的位置来识别所述一组交通信号灯的点亮颜色。

32、第三方面，本申请实施例提供一种计算机存储介质，该计算机存储介质存储多条指令，所述多条指令适于由处理器加载并执行如上所述的方法。

33、第四方面，本申请实施例提供一种电子设备，包括：

34、处理器，用以实现一条或一条以上指令；及

35、计算机存储介质，用以存储多条指令，所述多条指令适于由处理器加载并执行如上所述的方法。

36、实施本申请实施例至少具有如下有益效果：

37、在逆光场景下，可以识别交通信号灯，也可提升交通信号灯识别的准确率。

技术特征：

1.一种交通信号灯识别方法，其特征在于，所述方法包括：

2.如权利要求1所述的方法，其特征在于，所述提取所述分割图中点亮区域包括：

3.如权利要求2所述的方法，其特征在于，所述使用预设阈值提取所述转换后的分割图中的所述点亮区域包括：

4.如权利要求3所述的方法，其特征在于，所述方法还包括：

5.如权利要求1所述的方法，其特征在于，所述根据所述点亮区域在所述分割图中的位置来识别所述一组交通信号灯的点亮颜色包括：

6.如权利要求1所述的方法，其特征在于，所述检测到所述第一图像处于逆光场景，将所述第一图像中的第一区域分割出来得到分割图包括：

7.如权利要求1所述的方法，其特征在于，所述检测到所述第一图像处于逆光场景包括：

8.一种交通信号灯识别装置，其特征在于，所述装置包括：

9.一种计算机存储介质，其特征在于：该计算机存储介质存储多条指令，所述多条指令适于由处理器加载并执行如权利要求1-7任一项所述的方法。

10.一种电子设备，其特征在于，包括：

技术总结
本申请实施例提供一种交通信号灯识别方法，所述方法包括：获取包括交通信号灯的第一图像；检测到所述第一图像处于逆光场景，将所述第一图像中的第一区域分割出来得到分割图，其中所述第一区域为一组交通信号灯图像区域；提取所述分割图中点亮区域；根据所述点亮区域在所述分割图中的位置来识别所述一组交通信号灯的点亮颜色。本申请还提供一种交通信号灯识别装置、计算机存储介质及电子设备。采用本申请的实施例，可以在逆光场景下识别交通信号灯，也可提升交通信号灯识别的准确率。

技术研发人员：简士超,郭锦斌
受保护的技术使用者：鸿海精密工业股份有限公司
技术研发日：
技术公布日：2024/1/15