[发明专利]图像增强方法、装置及计算机可读存储介质

权利要求书

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

获取多组训练数据，每组训练数据包括第一图像和第二图像，其中，所述第二图像的图像质量优于所述第一图像的图像质量；

构建包括第一级生成式对抗网络和第二级生成式对抗网络的图像增强模型，利用所述多组训练数据对所述图像增强模型进行训练，获得训练好的图像增强模型，其中，以所述第二图像作为目标图像，基于所述第一图像的低频特征生成的增强低频图像，对所述第一级生成式对抗网络进行训练；以所述第二图像为目标图像，基于所述第一图像的融合图像生成的增强图像，对所述第二级生成式对抗网络进行训练；所述融合图像通过融合所述第一图像和增强低频图像得到；

将待增强的第三图像输入所述图像增强模型，输出图像增强后的第四图像。

2.根据权利要求1所述的图像增强方法，其特征在于，所述以所述第二图像作为目标图像，基于所述第一图像的低频特征生成的增强低频图像，对所述第一级生成式对抗网络进行训练，包括：

从所述第一图像中提取低频特征；

基于所述低频特征进行图像增强，生成所述第一图像的增强低频图像；

以所述第二图像作为目标图像，利用所述第一图像的增强低频图像对所述第一级生成式对抗网络进行训练，直至满足预设的训练结束条件。

3.根据权利要求2所述的图像增强方法，其特征在于，所述第一级生成式对抗网络的损失函数Loss_G_1为：Loss_G_1＝Lcobi₁+λ₁LG₁+η₁Lcolor₁；

其中，Lcobi₁表示针对所述增强低频图像和所述第二图像的上下文双边损失函数；LG₁表示针对所述增强低频图像与第二图像的对抗损失函数；Lcolor₁表示针对所述增强低频图像与第二图像的颜色损失函数；λ₁和η₁均为预设常数。

4.根据权利要求1至3任一项所述的图像增强方法，其特征在于，所述以所述第二图像为目标图像，基于所述第一图像的融合图像生成的增强图像，对所述第二级生成式对抗网络进行训练，包括：

对所述第一图像和增强低频图像相同位置的像素进行相加，得到融合图像，并基于所述融合图像，生成所述第一图像的增强图像；

以所述第二图像作为目标图像，利用所述第一图像的增强图像对所述第二级生成式对抗网络进行训练，直至满足预设的训练结束条件。

5.根据权利要求4所述的图像增强方法，其特征在于，所述第二级生成式对抗网络的损失函数Loss_G_2为：Loss_G_2＝η₂Lcobi-hf+η₃Lcobi₂+λ₂LG₂+η₄Lcolor₂；

其中，Lcobi₂表示针对所述增强图像和所述第二图像的上下文双边损失函数；Lcobi-hf表示针对所述增强图像的高频特征和所述第二图像的高频特征的上下文双边损失函数；LG₂表示针对所述增强图像与第二图像的对抗损失函数；Lcolor₂表示针对所述增强图像与第二图像的颜色损失函数；η₂、η₃、λ₂和η₄均为预设常数。

6.根据权利要求4所述的图像增强方法，其特征在于，同一组训练数据中的第一图像和第二图像所拍摄的内容相同。

7.根据权利要求1所述的图像增强方法，其特征在于，所述第一图像为等距柱状投影图或透视图，所述第二图像为透视图。