4 个月前

用于去除椒盐噪声的中值层卷积神经网络

用于去除椒盐噪声的中值层卷积神经网络

摘要

我们提出了一种具有新型层——中值层(median layer)的深度全卷积神经网络,用于恢复受到椒盐噪声(salt-and-pepper noise)污染的图像。中值层对所有特征通道简单地执行中值滤波操作。通过将这种类型的层添加到一些广泛使用的全卷积深度神经网络中,我们开发了一个端到端的网络,该网络能够在不进行任何复杂的预处理任务的情况下去除极高水平的椒盐噪声,这与现有的所有关于椒盐噪声去除的文献不同。实验结果表明,在简单的全卷积网络中插入中值层并使用L2损失函数可以显著提高信噪比。定量比较证明,我们的网络在有限的训练数据下优于现有最先进的方法。源代码已公开发布以供评估和使用(https://github.com/llmpass/medianDenoise)。

代码仓库

基准测试

基准方法指标
salt-and-pepper-noise-removal-on-bsd300-noiseCNN (Median Layers)
PSNR: 40.90
salt-and-pepper-noise-removal-on-bsd300-noise-1CNN (Median Layers)
PSNR: 37.28
salt-and-pepper-noise-removal-on-bsd300-noise-2CNN (Median Layers)
PSNR: 32.4
salt-and-pepper-noise-removal-on-kodak24-1CNN (Median Layers)
PSNR: 36.39
salt-and-pepper-noise-removal-on-kodak24-2CNN (Median Layers)
PSNR: 34.35
salt-and-pepper-noise-removal-on-kodak24-3CNN (Median Layers)
PSNR: 31.56

用 AI 构建 AI

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

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

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
用于去除椒盐噪声的中值层卷积神经网络 | 论文 | HyperAI超神经