本发明涉及图像处理,尤其涉及一种图像处理方法及装置。
背景技术:
1、注意力机制(attention mechanism)是一种用于在机器学习过程中自动学习和计算输入数据对输出数据的贡献大小的机制,由于注意力机制可以极大的提升图像处理模型的图像处理效果,因此用于实现注意力机制的注意力模块在图像超分、图像去模糊、图像分割、图像分类、图像识别等图像处理模型中的使用非常普遍。
2、由于在包含注意力模块的图像处理模型中,注意力模块一般需要多次被嵌入图像处理模型的不同位置,因此注意力模块的轻量化对图像处理模型的轻量化的影响非常大。然而,现有注意力模块普遍不够轻量化,进而导致包含注意力机制的图像处理模型在进行图像处理时的计算量较大。
技术实现思路
1、有鉴于此,本发明提供了一种图像处理方法及装置,用于减少包含注意力模块的图像处理模型在进行图像处理时的计算量。
2、为了实现上述目的,本发明实施例提供技术方案如下:
3、第一方面,本发明的实施例提供了一种图像处理方法,包括:
4、根据待处理图像获取第一图像特征;
5、对所述第一图像特征进行通道信息融合获取第二图像特征;
6、对所述第二图像特征进行空间信息融合获取第三图像特征;
7、融合所述第二图像特征和所述第三图像特征获取第四图像特征;
8、根据所述第四图像特征对所述第一图像特征进行标定,获取所述第一图像特征对应的重标定特征;
9、根据所述重标定特征获取所述待处理图像的处理结果。
10、第二方面,本发明实施例提供了一种图像处理装置,包括:
11、特征获取模块,用于根据待处理图像获取第一图像特征;
12、通道融合模块,用于对所述第一图像特征进行通道信息融合获取第二图像特征;
13、空间融合模块,对所述第二图像特征进行空间信息融合获取第三图像特征;
14、特征融合模块,用于融合所述第二图像特征和所述第三图像特征获取第四图像特征;
15、重标定模块,用于根据所述第四图像特征对所述第一图像特征进行标定,获取所述第一图像特征对应的重标定特征;
16、生成模块,用于根据所述重标定特征获取所述待处理图像的处理结果。
17、第三方面,本发明实施例提供了一种电子设备,包括:存储器和处理器,所述存储器用于存储计算机程序;所述处理器用于在调用计算机程序时,使得所述电子设备实现第一方面或第一方面任一种可选的实施方式所述的图像处理方法。
18、第四方面,本发明实施例提供一种计算机可读存储介质,当所述计算机程序被计算设备执行时,使得所述计算设备实现第一方面或第一方面任一种可选的实施方式所述的图像处理方法。
19、第五方面,本发明实施例提供一种计算机程序产品,当所述计算机程序产品在计算机上运行时,使得所述计算机实现第一方面或第一方面任一种可选的实施方式所述的图像处理方法。
20、本发明实施例中提供的图像处理方法在根据待处理图像获取第一图像特征后,首先对所述第一图像特征进行通道信息融合获取第二图像特征,然后对所述第二图像特征进行空间信息融合获取第三图像特征,再融合所述第二图像特征和所述第三图像特征获取第四图像特征,以及根据所述第四图像特征对所述第一图像特征进行标定,获取所述第一图像特征对应的重标定特征,最后根据所述重标定特征获取所述待处理图像的处理结果。由于本发明实施例是根据融合第二图像特征和第三图像特征得到的第四图像特征对所述第一图像特征进行标定的,而第二图像特征是进行通道信息融合得到的特征,第三图像特征是进行空间信息融合得到的特征,因此本发明实施例可以从通道信息和空间信息两个维度对第一图像特征进行重标定。又因为本发明实施例从通道信息和空间信息两个维度对第一图像特征进行重标定的过程中,无需分别独立对从通道信息和空间信息对第一图像特征进行重标定,因此本发明实施例可以减少注意力模块的计算量,进而减少包含注意力模块的图像处理模型在进行图像处理时的计算量。
1.一种图像处理方法,其特征在于,包括:
2.根据权利要求1所述的方法,其特征在于,所述对所述第一图像特征进行通道信息融合获取第二图像特征,包括:
3.根据权利要求1所述的方法,其特征在于,所述对所述第二图像特征进行空间信息融合获取第三图像特征,包括:
4.根据权利要求1所述的方法,其特征在于,所述对所述第二图像特征进行空间信息融合获取第三图像特征,包括:
5.根据权利要求3或4所述的方法,其特征在于,在通过第二卷积层对所述第二图像特征进行卷积操作获取第五图像特征之前,所述方法还包括:
6.根据权利要求1所述的方法,其特征在于,所述对所述第二图像特征进行空间信息融合获取第三图像特征,包括:
7.根据权利要求6所述的方法,其特征在于,在通过第二卷积层对所述第二图像特征进行卷积操作获取第五图像特征之前,所述方法还包括:
8.根据权利要求1所述的方法,其特征在于,所述融合所述第二图像特征和所述第三图像特征获取第四图像特征,包括:
9.根据权利要求1所述的方法,其特征在于,所述融合所述第二图像特征和所述第三图像特征获取第四图像特征,包括:
10.根据权利要求1所述的方法,其特征在于,所述根据所述第四图像特征对所述第一图像特征进行标定,获取所述第一图像特征对应的重标定特征,包括:
11.一种图像处理装置,其特征在于,包括:
12.一种电子设备,其特征在于,包括:存储器和处理器,所述存储器用于存储计算机程序;所述处理器用于在调用计算机程序时,使得所述电子设备实现权利要求1-10任一项所述的图像处理方法。
13.一种计算机可读存储介质,其特征在于,所述计算机可读存储介质上存储有计算机程序,当所述计算机程序被计算设备执行时,使得所述计算设备实现权利要求1-10任一项所述的图像处理方法。
14.一种计算机程序产品,其特征在于,当所述计算机程序产品在计算机上运行时,使得所述计算机实现如权利要求1-10任一项所述的图像处理方法。