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

权利要求书

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

获取输入图像；

将所述输入图像输入至U形网络中，得到输出图像；

其中，所述U形网络包括至少一个上采样单元、至少一个下采样单元以及混洗处理单元，所述混洗处理单元配置成：

将所述至少一个下采样单元输出的图像数据在至少一个变换域上进行处理，并将处理后的图像数据输入至所述至少一个上采样单元中，以使得所述至少一个上采样单元对接收到的图像数据进行处理。

2.根据权利要求1所述的方法，其特征在于，将所述至少一个下采样单元输出的图像数据在至少一个变换域上进行处理，包括：

将所述至少一个下采样单元输出的图像数据按照通道进行分组，得到分组后的图像数据；

按照预设的通道上的运算规则，对分组后的每组图像数据中的一部分图像数据在通道上进行运算处理，得到通道上运算后的数据。

3.根据权利要求2所述的方法，其特征在于，将所述至少一个下采样单元输出的图像数据在至少一个变换域上进行处理，包括：

按照预设的空间上的运算规则，对分组后的每组图像数据中的另一部分图像数据在空间上进行运算处理，得到空间上运算后的数据。

4.根据权利要求3所述的方法，其特征在于，将所述至少一个下采样单元输出的图像数据在至少一个变换域上进行处理，包括：

将所述每组图像数据中通道上运算后的数据和空间上运算后的数据进行拼接处理，得到处理后的每组图像数据；

将所述处理后的每组图像数据进行通道洗牌和拼接处理，得到所述处理后的图像数据。

5.根据权利要求1所述的方法，其特征在于，所述方法还包括：

基于预设的搜索空间，预设的目标函数，预设的性能评价客观指标以及预设的网络参数的优化方式进行自动网络搜索，得到一个下采样单元和一个上采样单元；

根据所述一个下采样单元和所述一个上采样单元，构建得到所述U形网络。

6.根据权利要求5所述的方法，其特征在于，所述根据所述一个下采样单元和所述一个上采样单元，构建得到所述U形网络，包括：

将所述一个下采样单元作为所述U形网络的下采样单元，将所述一个上采样单元作为所述U形网络的上采样单元，并在所述U形网络中至少一个下采样单元与至少一个上采样单元之间设置混洗处理单元，以构建得到所述U形网络。

7.根据权利要求5所述的方法，其特征在于，所述根据所述一个下采样单元和所述一个上采样单元，构建得到所述U形网络，包括：

将所述一个下采样单元作为所述U形网络的下采样单元，将所述一个上采样单元作为所述U形网络的上采样单元，并在所述U形网络中至少一个下采样单元与至少一个上采样单元之间设置混洗处理单元，得到初始的U形网络；

对所述初始的U形网络进行模型训练，得到训练后的U形网络，并将训练后的U形网络确定为所述U形网络。

8.根据权利要求5至7任一项所述的方法，其特征在于，所述预设的搜索空间包括：

四个网络节点和三种算子；其中，所述三种算子包括：上采样算子，下采样算子和恒等映射算子。

9.根据权利要求5至7任一项所述的方法，其特征在于，所述预设的目标函数为：

f＝argminL₁+2·L_ms-ssim+3·L_edge+L₂

其中，f为损失函数，L₁为网络输出与给定真值之间的1范数，L₂为网络输出与给定真值之间的2范数，L_ms-ssim为结构化损失，L_edge为所述输入图像的边缘信息损失。

10.根据权利要求5至7任一项所述的方法，其特征在于，所述预设的性能评价客观指标包括：

峰值信噪比和结构相似比。