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

权利要求书

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

获取多组成对样本数据和多组非成对样本数据，每组成对样本数据包括图像内容相同但图像质量不同的两个图像，每组非成对样本数据包括图像内容和图像质量均不同的两个图像，且所述多组非成对样本数据包括多个领域的图像；

基于所述多组成对样本数据和所述多组非成对样本数据，对图像增强模型进行训练，得到训练后的所述图像增强模型，所述成对样本数据用于使得训练后的所述图像增强模型学习到增强输入图像的图像质量的能力，所述非成对样本数据用于使得训练后的所述图像增强模型学习到适应多个领域的输入图像的能力；

在获取到待增强图像的情况下，通过训练后的所述图像增强模型对所述待增强图像进行图像增强，得到增强后的图像。

2.根据权利要求1所述的方法，其特征在于，所述每组成对样本数据包括第一图像和第二图像，且所述第二图像的图像质量高于所述第一图像，所述每组非成对样本数据包括第三图像和第四图像，且所述第四图像的图像质量高于所述第三图像；

所述基于所述多组成对样本数据和所述多组非成对样本数据，对图像增强模型进行训练，得到训练后的所述图像增强模型，包括：

对于任一组成对样本数据和任一组非成对样本数据，通过所述图像增强模型，对所述成对样本数据中的第一图像进行图像增强，得到第五图像；

通过所述图像增强模型，对所述非成对样本数据中的第三图像进行图像增强，得到第六图像；

基于所述第五图像和所述第二图像之间的差异，以及所述第六图像和所述第四图像之间的质量差异，训练所述图像增强模型，得到训练后的所述图像增强模型。

3.根据权利要求2所述的方法，其特征在于，所述通过所述图像增强模型，对所述成对样本数据中的第一图像进行图像增强，得到第五图像，包括：

调整所述第一图像的曝光度，得到曝光度为第一数值的第七图像和曝光度为第二数值的第八图像；

通过所述图像增强模型，对所述第七图像进行图像增强，得到第九图像；

通过所述图像增强模型，对所述第八图像进行图像增强，得到第十图像；

通过所述图像增强模型，将所述第九图像和所述第十图像进行融合，得到所述第五图像。

4.根据权利要求3所述的方法，其特征在于，所述通过所述图像增强模型，对所述第七图像进行图像增强，得到第九图像，包括：

通过所述图像增强模型，对所述第七图像进行特征提取，得到所述第七图像的第一特征图；

通过所述图像增强模型，对所述第一特征图进行全局信息和局部信息增强，得到第二特征图；

通过所述图像增强模型，对所述第二特征图进行解码，得到所述第九图像。

5.根据权利要求3所述的方法，其特征在于，所述通过所述图像增强模型，将所述第九图像和所述第十图像进行融合，得到所述第五图像，包括：

通过所述图像增强模型，对用于生成所述第九图像的第二特征图和用于生成所述第十图像的第三特征图进行特征提取，得到第四特征图，所述第四特征图用于表示相同区域内所述第九图像和所述第十图像的相对质量情况；

通过所述图像增强模型，基于所述第四特征图，将所述第九图像和所述第十图像进行融合，得到所述第五图像。

6.根据权利要求2所述的方法，其特征在于，所述通过所述图像增强模型，对所述非成对样本数据中的第三图像进行图像增强，得到第六图像，包括：

通过所述图像增强模型，对所述第三图像进行特征提取，得到所述第三图像的第五特征图；

通过所述图像增强模型，对所述第五特征图进行全局信息和局部信息增强，得到第六特征图；

通过所述图像增强模型，对所述第六特征图进行解码，得到所述第六图像。