[发明专利]一种图像处理方法、装置、电子设备和存储介质

说明书

本发明实施例公开了一种图像处理方法、装置、电子设备和存储介质。所述方法包括：获取行驶设备上安装的图像采集装置采集的视频流，从所述视频流中确定包含特定交通对象的多帧图像；确定所述多帧图像中的每帧图像中的特定交通对象的类别及其置信度；基于所述多帧图像中的特定交通对象的类别的置信度的比较结果，确定置信度不满足预设条件的类别的校正信息。

技术领域

本发明涉及图像处理技术领域，具体涉及一种图像处理方法、装置、电子设备和存储介质。

背景技术

目前，可通过多分类任务模型实现交通对象(例如交通标识)的分类识别，但由于交通对象众多，且容易受到远距离或物体遮挡等影响，因此存在识别准确率不高的问题。

发明内容

为解决现有存在的技术问题，本发明实施例提供一种图像处理方法、装置、电子设备和存储介质。

为达到上述目的，本发明实施例的技术方案是这样实现的：

本发明实施例提供了一种图像处理方法，所述方法包括：

获取行驶设备上安装的图像采集装置采集的视频流，从所述视频流中确定包含特定交通对象的多帧图像；

确定所述多帧图像中的每帧图像中的特定交通对象的类别及其置信度；

基于所述多帧图像中的特定交通对象的类别的置信度的比较结果，确定置信度不满足预设条件的类别的校正信息。

上述方案中，从所述视频流中确定包含特定交通对象的多帧图像，包括：

确定所述视频流中的包含交通对象的图像中的交通对象在图像中的第一区域；

针对每个第一区域，在所述第一区域内确定第二区域，所述第二区域小于所述第一区域；

基于各个第二区域的信息，从所述包含交通对象的图像中挑选包含第一类别的交通对象的图像作为包含特定交通对象的多帧图像，所述第一类别为所述特定交通对象所属的类别。

上述方案中，所述第二区域为所述第一区域的中心区域，所述第二区域的信息为所述第一区域的特征图的中心区域的信息。

上述方案中，所述基于各个第二区域的信息，从所述包含交通对象的图像中挑选包含第一类别的交通对象的图像，包括：

分别对所述各个第二区域进行特征提取，基于提取的特征确定所述各个第二区域的像素点的第一相似度；

将位置信息满足第一预设条件，且所述第一相似度满足第二预设条件的第二区域所在的图像确定为包含第一类别的交通对象的图像。

上述方案中，所述确定所述多帧图像中的每帧图像中的特定交通对象的类别及其置信度，包括：

确定所述多帧图像中的每帧图像中的第一类别的交通对象的置信度最大的细分类别及其置信度，所述第一类别为所述特定交通对象所属的类别。

上述方案中，所述基于所述多帧图像中的特定交通对象的类别的置信度的比较结果，确定置信度不满足预设条件的交通对象的校正信息，包括：

基于所述多帧图像中的所述第一类别的交通对象的置信度最大的细分类别的置信度的比较结果，确定最大置信度不满足预设条件的第一类别的交通对象的细分类别的校正信息。

上述方案中，所述确定所述多帧图像中的每帧图像中的第一类别的交通对象的置信度最大的细分类别及其置信度，包括：

确定所述多帧图像中的每帧图像中的第一类别的交通对象与各个第二类别的模板图像之间的第二相似度，各个第二类别为第一类别的细分类别；

基于所述第二相似度确定所述每帧图像中第一类别的交通对象的置信度最大的细分类别及其置信度。