[发明专利]图像处理方法、装置、电子设备及计算机可读存储介质

说明书

本申请实施例提供了一种图像处理方法、装置、电子设备及计算机可读存储介质，属于图像处理领域，该方法包括：基于待处理图像获取至少两个具有不同空间分辨率的特征图；基于至少两个特征图进行至少一次小波分析，得到具有目标空间分辨率的上采样特征图。基于本申请实施例提供的方案，能够结合至少两个具有不同空间分辨率的特征图的特征得到上采样特征图，从而使得到的上采样特征图具有更多的细节特征，图像更加清晰。

技术领域

本申请涉及图像处理技术领域，具体而言，本申请涉及一种图像处理方法、装置、电子设备及计算机可读存储介质。

背景技术

随着计算机视觉和人工智能的发展，在对图像进行处理时，如进行深度估计、语义分割和法向量估计时，需要对特征图进行上采样。特征图的上采样是指将低空间分辨率的特征图，提升到更高空间分辨率的特征图的过程，通过上采样算法可以获得更多细节的特征图。

通常对特征图进行上采样，需要获取具有目标空间分辨率的上采样特征图时，直接用具有低空间分辨率的特征图进行线性插值或反卷积，这种方式能够利用的信息就只有具有低空间分辨率的特征图本身，而此时的特征图中的细节信息已经非常少，导致上采样特征图难以获得更多的细节特征，即上采样特征图较模糊。

发明内容

本申请的目的旨在提供一种能够有效提高图像处理效果的图像处理方法、装置、电子设备及计算机可读存储介质。

第一方面，本申请实施例提供了一种图像处理方法，该方法包括：

基于待处理图像获取至少两个具有不同空间分辨率的特征图；

基于至少两个特征图进行至少一次小波分析，得到具有目标空间分辨率的上采样特征图。

第二方面，本申请实施例提供了一种图像处理装置，该装置包括：

提取模块，用于基于待处理图像获取至少两个具有不同空间分辨率的特征图；

分析模块，用于基于至少两个特征图进行至少一次小波分析，得到具有目标空间分辨率的上采样特征图。

第三方面，本申请实施例提供了一种电子设备，该电子设备包括处理器和存储器：

存储器，用于存储机器可读指令；

处理器，用于在执行上述机器可读指令时，执行上述第一方面所示的方法。

第四方面，本申请实施例提供了一种计算机可读存储介质，该存储介质上存储有计算机指令，计算机指令被处理器执行时实现上述第一方面所示的方法。

本申请提供的技术方案带来的有益效果是：

上述的图像处理方法，通过基于至少两个具有不同空间分辨率的特征图进行至少一次小波分析，得到具有目标空间分辨率的上采样特征图，得到的上采样特征图可以结合至少两个具有不同空间分辨率的特征图的特征，从而具有更多的细节特征，图像更加清晰。

进一步的，对第二特征图和由第一特征图进行小波分解得到的LH子带、HL子带和HH子带进行逆小波重构，可以实现一种直观的可解释的特征上采样，并且能够有效地增强上采样特征的细节。

进一步的，去掉通用的SSIM指标中的亮度指标，选用对比度和结构来度量SSIM相似度指标，得到注意力权重图，根据注意力权重图分别对LH子带、HL子带和HH子带进行特征变换，对第二特征图、变换LH子带、变换HL子带和变换HH子带进行逆小波重构，可以过滤掉第一特征图中与第二特征图不一致的高频信息，有效减轻由不同空间的特征图进行特征融合引起的纹理拷贝问题，使得上采样特征图更加准确，从而进一步提升模型的精度。

附图说明

为了更清楚地说明本申请实施例中的技术方案，下面将对本申请实施例描述中所需要使用的附图作简单地介绍。