
摘要
近年来,超深卷积神经网络(CNNs)在图像恢复领域受到了广泛关注。然而,随着网络深度的增加,这些超深模型中长期依赖问题却很少被重视,导致先前状态或层对后续层的影响微乎其微。受人类思维具有持续性的启发,我们提出了一种超深持续记忆网络(MemNet),该网络引入了一种记忆模块,由递归单元和门控单元构成,通过自适应学习过程显式挖掘持续记忆。递归单元在不同感受野下学习当前状态的多层级表示。这些表示与前序记忆模块的输出进行拼接,并输入至门控单元,由其自适应地控制应保留的先前状态比例,并决定当前状态应存储的信息量。我们将MemNet应用于三项图像恢复任务:图像去噪、超分辨率重建以及JPEG块效应去除。大量实验结果表明,MemNet具有必要性,并在所有三项任务上均显著优于现有最先进方法。代码已开源,地址为:https://github.com/tyshiwo/MemNet。
代码仓库
tyshiwo/MemNet
官方
tf
rshwndsz/denoiser
pytorch
GitHub 中提及
基准测试
| 基准 | 方法 | 指标 |
|---|---|---|
| color-image-denoising-on-cbsd68-sigma50 | MemNet | PSNR: 26.33 |
| image-super-resolution-on-bsd100-4x-upscaling | MemNet | PSNR: 27.40 SSIM: 0.7281 |
| image-super-resolution-on-manga109-4x | MemNet | PSNR: 29.42 SSIM: 0.8942 |
| image-super-resolution-on-set14-4x-upscaling | MemNet | PSNR: 28.26 SSIM: 0.7723 |
| image-super-resolution-on-urban100-4x | MemNet | PSNR: 25.50 SSIM: 0.7630 |
| jpeg-artifact-correction-on-classic5-quality | MemNet | PSNR: 29.69 |
| jpeg-artifact-correction-on-live1-quality-10 | MemNet | PSNR: 27.33 PSNR-B: 27.34 SSIM: 0.810 |
| jpeg-artifact-correction-on-live1-quality-10-1 | MemNet | PSNR: 29.45 PSNR-B: 29.39 SSIM: 0.8327 |
| jpeg-artifact-correction-on-live1-quality-20 | MemNet | PSNR: 29.76 PSNR-B: 29.75 SSIM: 0.877 |
| jpeg-artifact-correction-on-live1-quality-20-1 | MemNet | PSNR: 31.83 PSNR-B: 31.74 SSIM: 0.8970 |