中文

基于注意力与对齐网络的高效渐进高动态范围图像复原

计算机视觉与模式识别 2022-04-21 v1 机器学习 图像与视频处理

摘要

高动态范围(HDR)是计算摄影技术的重要组成部分。本文中,我们提出了一种称为高效注意力与对齐引导渐进网络(EAPNet)的轻量神经网络,用于应对NTIRE 2022 HDR Track 1与Track 2挑战赛。我们引入了多维轻量编码模块来提取特征。此外,我们提出了渐进空洞U形块(PDUB),其可作为动态调节MAccs与PSNR的渐进即插即用模块。最后,我们使用快速低功耗特征对齐模块处理错位问题,以替代耗时的可变形卷积网络(DCN)。实验表明,与最先进方法相比,我们的方法在MAccs上实现了约20倍压缩,且mu-PSNR与PSNR更优。我们在测试阶段获得了两个赛道的第二名。图1展示了NTIRE 2022 HDR挑战赛的可视化结果。

关键词

引用

@article{arxiv.2204.09213,
  title  = {Efficient Progressive High Dynamic Range Image Restoration via Attention and Alignment Network},
  author = {Gaocheng Yu and Jin Zhang and Zhe Ma and Hongbin Wang},
  journal= {arXiv preprint arXiv:2204.09213},
  year   = {2022}
}