[发明专利]图像处理方法、图像处理装置和存储介质

权利要求书

1.一种图像处理方法，包括：

获取输入图像；

利用图像重建网络对所述输入图像进行N次重建操作，以得到所述输入图像对应的重建图像，

其中，N为大于等于1的正整数，

所述图像重建网络包括编码子网络、解码子网络和连接子网络，

所述编码子网络包括至少一个编码块，每个编码块包括多个自注意力算子，

所述解码子网络包括至少一个解码块，每个解码块包括多个自注意力算子，

所述连接子网络包括连接层，

所述至少一个编码块中的一个编码块与所述至少一个解码块中的一个解码块通过所述连接层映射连接，

其中，每次重建操作包括：

通过所述编码子网络对所述编码子网络的输入进行特征提取处理，以得到第一特征图和所述至少一个编码块的输出；

利用所述解码子网络和所述连接子网络对所述第一特征图和所述至少一个编码块的输出进行处理，以得到所述解码子网络的输出。

2.根据权利要求1所述的图像处理方法，其中，所述图像重建网络还包括瓶颈层，

其中，在所述图像重建网络包括所述瓶颈层的情况下，每次重建操作包括：

通过所述编码子网络对所述编码子网络的输入进行特征提取处理，以得到所述第一特征图和所述至少一个编码块的输出；

利用所述瓶颈层对所述第一特征图进行处理，以得到第二特征图；

利用所述解码子网络和所述连接子网络对所述第二特征图和所述至少一个编码块的输出进行处理，以得到所述解码子网络的输出。

3.根据权利要求2所述的图像处理方法，其中，所述瓶颈层包括多个自注意力算子，

利用所述瓶颈层对所述第一特征图进行处理，以得到第二特征图，包括：

利用所述瓶颈层的多个自注意力算子对所述第一特征图进行表示学习，以得到所述第二特征图。

4.根据权利要求3所述的图像处理方法，其中，所述瓶颈层的多个自注意力算子包括依次交替连接的基于窗口的自注意力算子和基于移动窗口的自注意力算子，

所述瓶颈层中的基于窗口的自注意力算子的数量比基于移动窗口的自注意力算子的数量多1。

5.根据权利要求1-4中任一项所述的图像处理方法，其中，

当N为1，对于所述N次重建操作，所述编码子网络的输入为所述输入图像，所述解码子网络的输出为所述重建图像，

当N大于1，对所述N次重建操作中的第i次重建操作中所述解码子网络的输出进行数据拟合处理和数据修正处理，以得到所述N次重建操作中的第i+1次重建操作中所述编码子网络的输入，i为正整数且小于N，

i为1时，所述第i次重建操作中所述编码子网络的输入为所述输入图像，i为N-1时，所述第i+1次重建操作中所述解码子网络的输出为所述重建图像。

6.根据权利要求2-4中任一项所述的图像处理方法，其中，所述连接子网络包括至少一个连接层，所述至少一个编码块与所述至少一个解码块通过所述至少一个连接层一一对应映射连接，

每个所述连接层用于将具有相同尺寸特征的编码块和解码块进行映射连接。

7.根据权利要求1-4中任一项所述的图像处理方法，其中，所述解码子网络还包括线性投影层，所述至少一个编码块包括依次连接的多个编码块，所述至少一个解码块包括依次连接的多个解码块，所述连接子网络包括多个连接层，

所述多个编码块与所述多个解码块通过所述多个连接层一一对应映射连接，

每个所述连接层用于将所述连接层对应的编码块的输出和所述连接层对应的解码块的输出进行连接处理，

所述连接层的输出作为所述连接层对应的解码块所连接的下一个解码块的输入或者所述连接层的输出作为所述线性投影层的输入。