17 天前

FFDNet:面向基于CNN图像去噪的快速灵活解决方案

FFDNet:面向基于CNN图像去噪的快速灵活解决方案

摘要

由于具备快速推理能力和良好的性能表现,判别式学习方法在图像去噪领域得到了广泛研究。然而,这类方法通常针对每种特定的噪声水平训练一个专用模型,对于不同噪声水平的图像去噪需要多个模型,且难以应对空间变化的噪声,限制了其在实际去噪应用中的灵活性与适用性。为解决上述问题,本文提出一种快速且灵活的去噪卷积神经网络——FFDNet,其输入为可调的噪声水平图(noise level map)。FFDNet在下采样后的子图像上进行处理,实现了推理速度与去噪性能之间的良好平衡。与现有的判别式去噪方法相比,FFDNet具有多项优越特性:(i)仅需一个网络即可有效处理广泛的噪声水平范围(即[0, 75]);(ii)通过指定非均匀的噪声水平图,能够有效去除空间变化的噪声;(iii)即使在CPU上运行,其推理速度也快于基准方法BM3D,且不牺牲去噪性能。我们在合成图像和真实噪声图像上进行了大量实验,将FFDNet与当前最先进的去噪方法进行对比。实验结果表明,FFDNet在去噪效果与效率方面均表现优异,具有很高的实用价值,非常适用于实际的图像去噪应用场景。

代码仓库

cszn/FFDNet
官方
pytorch
GitHub 中提及
7568/ffdnet-pytorch
pytorch
GitHub 中提及
SamirMitha/Denoising
tf
GitHub 中提及
Aoi-hosizora/FFDNet_pytorch
pytorch
GitHub 中提及
mq0829/DL-CACTI
tf
GitHub 中提及
LucasElbert/FFDNet
pytorch
GitHub 中提及
deshanyang/liver-dir-qa
GitHub 中提及

基准测试

基准方法指标
color-image-denoising-on-cbsd68-sigma15FFDNet
PSNR: 33.87
color-image-denoising-on-cbsd68-sigma25FFDNet
PSNR: 31.21
color-image-denoising-on-cbsd68-sigma35FFDNet
PSNR: 29.58
color-image-denoising-on-cbsd68-sigma50FFDNet
PSNR: 27.96
color-image-denoising-on-cbsd68-sigma75FFDNet
PSNR: 26.24
color-image-denoising-on-kodak25-sigma15FFDNet
PSNR: 34.63
color-image-denoising-on-kodak25-sigma25FFDNet
PSNR: 32.13
color-image-denoising-on-kodak25-sigma35FFDNet
PSNR: 30.57
color-image-denoising-on-kodak25-sigma50FFDNet
PSNR: 28.98
color-image-denoising-on-kodak25-sigma75FFDNet
PSNR: 27.27
color-image-denoising-on-mcmaster-sigma15FFDNet
PSNR: 34.66
color-image-denoising-on-mcmaster-sigma25FFDNet
PSNR: 32.35
color-image-denoising-on-mcmaster-sigma35FFDNet
PSNR: 30.81
color-image-denoising-on-mcmaster-sigma50FFDNet
PSNR: 29.18
color-image-denoising-on-mcmaster-sigma75FFDNet
PSNR: 27.33
color-image-denoising-on-urban100-sigma15-1FFDNet
Average PSNR: 33.83
denoising-on-darmstadt-noise-datasetFFDNet
PSNR: 34.40
grayscale-image-denoising-on-bsd68-sigma15FFDNet
PSNR: 31.63
grayscale-image-denoising-on-bsd68-sigma25FFDNet
PSNR: 29.19
grayscale-image-denoising-on-bsd68-sigma35FFDNet
PSNR: 27.73
grayscale-image-denoising-on-bsd68-sigma50FFDNet
PSNR: 26.29
grayscale-image-denoising-on-bsd68-sigma75FFDNet
PSNR: 24.79
grayscale-image-denoising-on-clip300-sigma15FFDNet-Clip
PSNR: 31.68
grayscale-image-denoising-on-clip300-sigma25FFDNet-Clip
PSNR: 29.25
grayscale-image-denoising-on-clip300-sigma35FFDNet-Clip
PSNR: 27.75
grayscale-image-denoising-on-clip300-sigma50FFDNet-Clip
PSNR: 26.25
grayscale-image-denoising-on-clip300-sigma60FFDNet-Clip
PSNR: 25.51
grayscale-image-denoising-on-set12-sigma15FFDNet
PSNR: 25.49

用 AI 构建 AI

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

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

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
FFDNet:面向基于CNN图像去噪的快速灵活解决方案 | 论文 | HyperAI超神经