中文

基于质量注意力生成对抗网络的非配对图像增强

图像与视频处理 2021-01-01 v1 计算机视觉与模式识别

摘要

在这项工作中,我们旨在学习一个非配对图像增强模型,该模型可利用用户提供的高质量图像的特征来丰富低质量图像。我们提出了一种基于非配对数据训练的质量注意力生成对抗网络(QAGAN),其构建于嵌入质量注意力模块(QAM)的双向生成对抗网络(GAN)之上。所提 QAGAN 的关键新颖之处在于为生成器注入了 QAM,使其直接从两个域中学习域相关的质量注意力。更具体地,所提 QAM 使生成器能够分别从空间维度有效选择语义相关特征,并从通道维度自适应地融入风格相关属性。因此,在我们提出的 QAGAN 中,不仅判别器,生成器也能直接访问两个域,这极大促进了生成器对映射函数的学习。大量实验结果表明,基于非配对学习的现有最优方法相比,我们提出的方法在客观与主观评价中均取得更好性能。

关键词

引用

@article{arxiv.2012.15052,
  title  = {Unpaired Image Enhancement with Quality-Attention Generative Adversarial Network},
  author = {Zhangkai Ni and Wenhan Yang and Shiqi Wang and Lin Ma and Sam Kwong},
  journal= {arXiv preprint arXiv:2012.15052},
  year   = {2021}
}