[发明专利]图像数据挖掘方法、装置、设备及路侧设备

说明书

本公开提供了一种图像数据挖掘方法，涉及智能交通和自动驾驶领域，具体地，涉及计算机视觉或图像处理领域。该图像数据挖掘方法包括获取包含交通灯的图像；在图像中确定包括第一颜色的灯头在内的第一区域、包括第二颜色的灯头在内的第二区域和包括目标颜色的灯头在内的目标区域；分别获取表示目标区域与第一区域之间的差异的第一差分值、表示目标区域与第二区域之间的差异的第二差分值以及表示第一区域与第二区域之间的差异的第三差分值；根据第一差分值、第二差分值和第三差分值确定图像是否为目标图像，并存储目标图像作为图像数据挖掘的结果图像数据。本公开还提供了一种图像数据挖掘装置、设备、存储介质和路侧设备。

技术领域

本公开涉及智能交通或自动驾驶领域，具体地，涉及计算机视觉或图像处理领域，更具体地，涉及一种图像数据挖掘方法、装置、设备、存储介质及路侧设备。

背景技术

在智能交通中，通过使用路侧传感器和车载传感器相结合的车路协同技术来对行驶环境进行感测。路侧传感器架设在路边，以固定不变的感知区域采集图像数据。可以利用路侧传感器所采集的图像数据构建训练数据集，以训练并建立神经网络模型来对交通灯的颜色进行灯色识别。然而，由于黄灯数据在训练数据集中占比较少，因此会影响所建立的神经网络模型的识别效果，进而导致黄色交通灯识别失败。

发明内容

有鉴于此，本公开提供了一种图像数据挖掘方法、装置、设备及存储介质。

本公开的第一方面提供了一种图像数据挖掘方法，包括：

获取包含交通灯的图像，所述交通灯包括分别具有第一颜色、第二颜色和目标颜色的多个灯头；

在所述图像中确定包括所述第一颜色的灯头在内的第一区域、包括所述第二颜色的灯头在内的第二区域和包括所述目标颜色的灯头在内的目标区域，其中，所述第一区域、所述第二区域和所述目标区域具有彼此相同的形状和尺寸；

分别获取表示所述目标区域与所述第一区域之间的差异的第一差分值、表示所述目标区域与所述第二区域之间的差异的第二差分值以及表示所述第一区域与所述第二区域之间的差异的第三差分值；以及

根据所述第一差分值、所述第二差分值和所述第三差分值确定所述图像是否为目标图像，并且在确定所述图像为目标图像的情况下，存储所述目标图像作为图像数据挖掘的结果图像数据。

本公开的第二方面提供了一种图像数据挖掘方法，包括：

获取连续捕获的包含交通灯的第一帧图像和第二帧图像，所述交通灯包括分别具有第一颜色、第二颜色和目标颜色的多个灯头；

根据所述第一帧图像与所述第二帧图像之间沿第一差分方向的帧间差分来确定所述目标颜色的灯头是否被点亮；

在确定所述目标颜色的灯头被点亮的情况下，将所述目标颜色的灯头的状态设置为点亮；

根据所述第一帧图像与所述第二帧图像之间沿第二差分方向的帧间差分来确定所述目标颜色的灯头是否被熄灭；

在确定所述目标颜色的灯头被熄灭的情况下，将所述目标颜色的灯头的状态设置为熄灭；

在所述目标颜色的灯头的状态为点亮的情况下，根据所述第二帧图像中包括所述第一颜色的灯头在内的第一区域、包括所述第二颜色的灯头在内的第二区域和包括所述目标颜色的灯头在内的第一目标区域之间的区域间差分来确定所述第二帧图像是否为目标图像；以及

在确定所述第二帧图像为目标图像的情况下，存储所述目标图像作为图像数据挖掘的结果图像数据。

本公开的第三方面提供了一种图像数据挖掘装置，包括：

图像获取模块，配置为获取包含交通灯的图像，所述交通灯包括分别具有第一颜色、第二颜色和目标颜色的多个灯头；