[发明专利]图像增强方法、装置、计算机设备和存储介质

说明书

本申请涉及一种图像增强方法、装置、计算机设备和存储介质，本申请实施例可应用于云技术、人工智能、智慧交通、辅助驾驶等各种场景。所述方法包括：获取待处理图像；对待处理图像进行多尺度特征提取，得到多尺度图像特征；多尺度图像特征包括至少两个层级尺度图像特征；对多尺度图像特征中第一层级尺度图像特征进行注意力机制加权，得到第一图像特征，注意力机制包括空间注意力机制和通道注意力机制；对第一图像特征和第二层级尺度图像特征进行加权融合，得到第二图像特征，第二层级尺度图像特征为多尺度图像特征中非第一层级尺度图像特征；对第二图像特征进行特征提取，得到增强图像。采用本方法能够得到图像增强效果好的增强图像。

技术领域

本申请涉及图像处理技术领域，特别是涉及一种图像增强方法、装置、计算机设备、存储介质和计算机程序产品。

背景技术

随着图像处理技术的发展，出现了图像增强技术，图像增强技术主要用于对图像进行处理，以提升图像质量。比如，图像增强具体可以是指图像去雾，通过去雾来提升图像质量。

传统技术中，常用的图像增强方式为，训练深度神经网络，利用训练好的深度神经网络对待处理图像进行处理，以实现图像增强。

然而，传统方法，所使用的深度神经网络并不能充分提取待处理图像的特征，存在图像增强效果差的问题。

发明内容

基于此，有必要针对上述技术问题，提供一种能够提升图像增强效果的图像增强方法、装置、计算机设备、计算机可读存储介质和计算机程序产品。

第一方面，本申请提供了一种图像增强方法。所述方法包括：

获取待处理图像；

对待处理图像进行多尺度特征提取，得到多尺度图像特征；多尺度图像特征包括至少两个层级尺度图像特征；

对多尺度图像特征中第一层级尺度图像特征进行注意力机制加权，得到第一图像特征，注意力机制包括空间注意力机制和通道注意力机制；

对第一图像特征和第二层级尺度图像特征进行加权融合，得到第二图像特征，第二层级尺度图像特征为多尺度图像特征中非第一层级尺度图像特征；

对第二图像特征进行特征提取，得到增强图像。

第二方面，本申请还提供了一种图像增强装置。所述装置包括：

获取模块，用于获取待处理图像；

多尺度特征提取模块，用于对待处理图像进行多尺度特征提取，得到多尺度图像特征；多尺度图像特征包括至少两个层级尺度图像特征；

注意力机制加权模块，用于对多尺度图像特征中第一层级尺度图像特征进行注意力机制加权，得到第一图像特征，注意力机制包括空间注意力机制和通道注意力机制；

特征加权融合模块，用于对第一图像特征和第二层级尺度图像特征进行加权融合，得到第二图像特征，第二层级尺度图像特征为多尺度图像特征中非第一层级尺度图像特征；

处理模块，用于对第二图像特征进行特征提取，得到增强图像。

第三方面，本申请还提供了一种计算机设备。所述计算机设备包括存储器和处理器，所述存储器存储有计算机程序，所述处理器执行所述计算机程序时实现以下步骤：

获取待处理图像；

对待处理图像进行多尺度特征提取，得到多尺度图像特征；多尺度图像特征包括至少两个层级尺度图像特征

对多尺度图像特征中第一层级尺度图像特征进行注意力机制加权，得到第一图像特征，注意力机制包括空间注意力机制和通道注意力机制；