基于残差自注意力和生成对抗网络的图像增强方法

作者:秦翰林; 姚迪; 延翔; 马琳; 梁毅; 曾庆杰; 杨硕闻; 乐阳; 张嘉伟; 侯本照; 周慧鑫
来源:2020-12-02, 中国, ZL202011392344.3.

摘要

本发明公开了一种基于残差自注意力和生成对抗网络的图像增强方法,通过残差自注意力模块对低质量红外图像进行红外图像特征提取,获得低级红外特征图;通过增强模块对提取到的低级红外特征图进行多尺度深层次细节特征提取,获得高级红外特征图;低级特征图与高级特征图进行卷积合并生成增强红外图像,将增强红外图像和与低质量红外图像配对的高质量红外图像输入判别器,在生成对抗网络的生成器和判别器的训练过程中,确定两幅图像的纹理损失、内容损失、全变分损失,对比度损失,引导生成器与判别器的参数更新方向,不断提高增强红外图像的细节特征,获得质量相对最优的增强红外图像。