17 天前

MemNet:一种用于图像恢复的持久化记忆网络

MemNet:一种用于图像恢复的持久化记忆网络

摘要

近年来,超深卷积神经网络(CNNs)在图像恢复领域受到了广泛关注。然而,随着网络深度的增加,这些超深模型中长期依赖问题却很少被重视,导致先前状态或层对后续层的影响微乎其微。受人类思维具有持续性的启发,我们提出了一种超深持续记忆网络(MemNet),该网络引入了一种记忆模块,由递归单元和门控单元构成,通过自适应学习过程显式挖掘持续记忆。递归单元在不同感受野下学习当前状态的多层级表示。这些表示与前序记忆模块的输出进行拼接,并输入至门控单元,由其自适应地控制应保留的先前状态比例,并决定当前状态应存储的信息量。我们将MemNet应用于三项图像恢复任务:图像去噪、超分辨率重建以及JPEG块效应去除。大量实验结果表明,MemNet具有必要性,并在所有三项任务上均显著优于现有最先进方法。代码已开源,地址为:https://github.com/tyshiwo/MemNet。

代码仓库

rshwndsz/denoiser
pytorch
GitHub 中提及

基准测试

基准方法指标
color-image-denoising-on-cbsd68-sigma50MemNet
PSNR: 26.33
image-super-resolution-on-bsd100-4x-upscalingMemNet
PSNR: 27.40
SSIM: 0.7281
image-super-resolution-on-manga109-4xMemNet
PSNR: 29.42
SSIM: 0.8942
image-super-resolution-on-set14-4x-upscalingMemNet
PSNR: 28.26
SSIM: 0.7723
image-super-resolution-on-urban100-4xMemNet
PSNR: 25.50
SSIM: 0.7630
jpeg-artifact-correction-on-classic5-qualityMemNet
PSNR: 29.69
jpeg-artifact-correction-on-live1-quality-10MemNet
PSNR: 27.33
PSNR-B: 27.34
SSIM: 0.810
jpeg-artifact-correction-on-live1-quality-10-1MemNet
PSNR: 29.45
PSNR-B: 29.39
SSIM: 0.8327
jpeg-artifact-correction-on-live1-quality-20MemNet
PSNR: 29.76
PSNR-B: 29.75
SSIM: 0.877
jpeg-artifact-correction-on-live1-quality-20-1MemNet
PSNR: 31.83
PSNR-B: 31.74
SSIM: 0.8970

用 AI 构建 AI

从想法到上线——通过免费 AI 协同编程、开箱即用的环境和市场最优价格的 GPU 加速您的 AI 开发

AI 协同编程
即用型 GPU
最优价格
立即开始

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
MemNet:一种用于图像恢复的持久化记忆网络 | 论文 | HyperAI超神经