一种基于注意力机制的生成对抗网络的超分辨率图像重建方法

摘要

本发明公开了一种基于注意力机制的生成对抗网络的超分辨率图像重建方法，包括：预处理ImageNet数据集，制作高、低分辨率图像对应的训练数据集；构建用于训练的生成对抗网络模型，在该模型中引入注意力机制；将得到的训练数据集依次输入生成对抗网络中进行模型训练；将待处理的图像输入到训练完毕的生成网络模型中，得到重建后的高分辨率图像。本发明通过在感知网络中加入注意力机制来提取目标的显著性区域，利用局部信息和全局信息结合的方式使生成的图像更加接近于真实的高分辨率图像，并引入感知损失提高生成效果，使重建后的图像边缘和细节信息更加清晰，重建效果更佳。