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

权利要求书

1.一种图像处理方法，其特征在于，所述方法包括：

基于第一算子将第一待处理图像的图像数据从第一数据排布方式转换至第二数据排布方式，以得到分辨率压缩的第二待处理图像；所述第一数据排布方式和所述第二数据排布方式在深度维度上的分量为通道数量，所述第一数据排布方式对应的通道数量小于所述第二数据排布方式对应的通道数量；

通过深度估计网络对所述第二待处理图像进行深度估计，得到与所述第二待处理图像对应的第一深度估计图像，所述第一深度估计图像的分辨率与所述第二待处理图像的分辨率一致；

基于第二算子将所述第一深度估计图像的图像数据从所述第二数据排布方式转换至所述第一数据排布方式，以得到分辨率与所述第一待处理图像一致的第二深度估计图像。

2.根据权利要求1所述的方法，其特征在于，所述第二待处理图像的分辨率为所述第一待处理图像的分辨率的二分之一。

3.根据权利要求1所述的方法，其特征在于，所述深度估计网络包括：编码器和解码器，所述解码器包括注意力模块；以及，所述通过训练好的深度估计网络对所述第二待处理图像进行深度估计，得到与所述第二待处理图像对应的第一深度估计图像，包括：

通过所述编码器对所述第二待处理图像进行编码处理，得到编码特征图像；

通过所述解码器中所述注意力模块，基于注意力机制对所述编码特征图像进行深度估计，以得到与所述第二待处理图像对应的第一深度估计图像。

4.根据权利要求3所述的方法，其特征在于，所述解码器包括：依次连接的至少两个所述注意力模块；每个所述注意力模块包括：网络层、第一映射函数和乘法算子；以及，所述通过所述解码器中所述注意力模块，基于注意力机制对所述编码特征图像进行深度估计，以得到所述注意力模块输出的第一深度估计图像，包括：

通过第M个注意力模块的网络层，对输入至所述第M个注意力模块的输入图像进行解码处理，得到第一原始特征图像，并通过所述第M个注意力模块中的第一映射函数对所述第一原始特征图像进行映射，再通过所述第M个注意力模块中的乘法算子将映射后得到的第一相关性特征图像与所述第一原始特征图像进行相乘，得到所述第M个注意力模块输出的解码特征图像；

其中，M为大于或等于1的正整数；当M等于1时，所述输入图像为所述编码器输出的所述编码特征图像；当M大于1时，所述输入图像为第M-1个注意力模块输出的解码特征图像；所述第一深度估计图像为所述解码器包括的最后一个注意力模块输出的解码特征图像，不同的所述注意力模块输出的解码特征图像的分辨率不同。

5.根据权利要求1-4任一项所述的方法，其特征在于，所述基于第二算子将所述第一深度估计图像的图像数据从所述第二数据排布方式转换至所述第一数据排布方式，以将所述第一深度估计图像的分辨率恢复至与所述第一待处理图像的分辨率一致，得到第二深度估计图像，包括：

基于第二算子将所述第一深度估计图像的图像数据从所述第二数据排布方式转换至所述第一数据排布方式，以将所述第一深度估计图像的分辨率恢复至与所述第一待处理图像的分辨率一致；

通过第二映射函数对分辨率恢复后的第一深度估计图像进行映射，将映射后得到的图像确定为第二深度估计图像。

6.一种根据权利要求1所述的方法，其特征在于，在得到所述第二深度估计图像之后，所述方法还包括：

确定所述第一待处理图像的背景区域，并根据所述第二深度估计图像确定所述背景区域对应的深度估计值；

根据所述背景区域对应的深度估计值确定所述背景区域的虚化等级，并按照所述虚化等级对所述背景区域进行虚化处理。