使用深度学习进行图像去噪(二)

简介: 使用深度学习进行图像去噪

传统图像去噪滤波器概述

传统上,研究人员想出了滤波器器来对图像进行降噪。大多数滤波器器特定于图像所具有的噪声类型。有几种类型的噪声,例如高斯噪声,泊松噪声,斑点噪声,椒盐(脉冲)噪声等。每种类型的噪声都有特定的滤波器。因此,使用传统滤波器对图像进行降噪的第一步是识别图像中存在的噪声类型。确定后,我们可以继续应用特定的滤波器器。为了识别噪声的类型,有一些数学公式可以帮助我们猜测噪声的类型。否则,领域专家可以仅通过查看图像来决定。还有一些滤波器可以处理任何类型的噪声。

有大量的滤波器可用于对图像进行降噪。每个人都有其优点和缺点。在这里,我将讨论非局部均值(NLM)算法,该算法被认为可以很好地对图像进行去噪。

NLM的公式,

640.png

该算法将像素的估计值计算为图像中所有像素的加权平均值,但是权重族取决于像素i和j之间的相似度。换句话说,它查看一个图像块,然后识别整个图像中的其他相似块,并对它们进行加权平均。要了解这一点,请考虑以下图像,

640.png

相似的色块用相同颜色的方框标记。因此,现在,它将相似补丁的像素的加权平均值作为目标像素的估计值。该算法将色块大小和色块距离作为输入。

考虑以下使用NLM滤镜去噪的灰度图像。

640.png

您可以看到NLM在图像去噪方面做得不错。如果仔细观察,将会发现去噪图像略有模糊。这是由于应用于任何数据的均值将使值平滑。

但是,当噪声水平太高时,NLM无法提供良好的结果。考虑以下图像,该图像已使用NLM滤波器进行了去噪。

640.png

可以清楚地看到,去噪后的图像太模糊了,大部分关键细节都丢失了。例如,观察蓝色卡车的橙色前灯。

用于图像去噪的深度学习模型

随着深度学习技术的出现,现在可以从图像中去除盲目的噪声,这样的结果非常接近于真实图像的细节损失最小。

已经实现了三个深度学习架构,

REDNet、MWCNN、PRIDNet

REDNet  -Residual Encoder-Decoder Networks

这是一个基于CNN的跳过连接的自动编码器架构。体系结构如下:

640.png

在这里,我用了5层卷积的编码器和5层反卷积的解码器。这是一个非常简单的体系结构,我将其作为基准。

input_0=Input(shape=(256,256,3), name="input_layer")
conv_layer_1=Conv2D(filters=256, kernel_size=2, padding='same', name="conv_1")(input_0)
conv_layer_2=Conv2D(filters=256, kernel_size=2, padding='same', name="conv_2")(conv_layer_1)
conv_layer_3=Conv2D(filters=256, kernel_size=3, padding='same', name="conv_3")(conv_layer_2)
conv_layer_4=Conv2D(filters=256, kernel_size=3, padding='same', name="conv_4")(conv_layer_3)
conv_layer_5=Conv2D(filters=128, kernel_size=3, padding='same', name="conv_5")(conv_layer_4)
deconv_layer_5=Conv2DTranspose(filters=256, kernel_size=2, padding='same', name="deconv_5")(conv_layer_5)
deconv_layer_5=Add(name="add_1")([conv_layer_4, deconv_layer_5])
deconv_layer_4=Conv2DTranspose(filters=256, kernel_size=2, padding='same', name="deconv_4")(deconv_layer_5)
deconv_layer_3=Conv2DTranspose(filters=256, kernel_size=3, padding='same', name="deconv_3")(deconv_layer_4)
deconv_layer_3=Add(name="add_2")([conv_layer_2, deconv_layer_3])
deconv_layer_2=Conv2DTranspose(filters=128, kernel_size=3, padding='same', name="deconv_2")(deconv_layer_3)
deconv_layer_1=Conv2DTranspose(filters=3, kernel_size=3, padding='same', name="deconv_1")(deconv_layer_2)
out=Add(name="add_3")([input_0, deconv_layer_1])
model=Model(inputs=[input_0], outputs=[out])

640.png

640.png

如您所见,该体系结构在去噪图像方面效果很好。您绝对可以看到噪点有所减少,并且图像正在尝试针对损坏的像素适应图像的原始颜色。该体系结构的PSNR得分为30.5713,SSIM得分为0.7932。

MWCNN — Multi-level Wavelet CNN

这是基于小波的深度学习架构。它的架构与U-Net架构有着惊人的相似性。MWCNN的唯一区别在于,与U-Net中的下采样和上采样不同,这里我们使用DWT(离散小波变换)和IWT(逆小波变换)。DWT和IWT的工作方式已超出此文章的范围。但是,我在[参考资料部分]附加了一些资源,您可以从中学习这些资源。

640.png

在这里,我已将此体系结构扩展到4个级别。因此,我的网络深度变为32。此代码有点长,我在Keras中使用了自定义层。您可以在Github存储库中查看有关MWCNN的完整代码。

640.png

640.png

640.png

我们可以看到,与REDNet相比,该架构的工作方式更好,图像更清晰。该体系结构的PSNR得分为32.5221,SSIM得分为0.8397。

目录
相关文章
|
4天前
|
机器学习/深度学习 监控 算法
深度学习之图像去噪与去模糊
基于深度学习的图像去噪和去模糊是计算机视觉中的重要任务,旨在提升图像质量,去除噪声和模糊。
23 3
|
机器学习/深度学习
使用深度学习进行图像去噪(三)
使用深度学习进行图像去噪
817 0
使用深度学习进行图像去噪(三)
|
机器学习/深度学习 编解码 算法
使用深度学习进行图像去噪(一)
使用深度学习进行图像去噪
641 0
使用深度学习进行图像去噪(一)
|
机器学习/深度学习 编解码 算法
使用深度学习进行图像去噪
使用深度学习进行图像去噪
652 0
使用深度学习进行图像去噪
|
3天前
|
机器学习/深度学习 数据采集 自然语言处理
深度学习在自然语言处理中的应用与挑战
本文探讨了深度学习技术在自然语言处理(NLP)领域的应用,包括机器翻译、情感分析和文本生成等方面。同时,讨论了数据质量、模型复杂性和伦理问题等挑战,并提出了未来的研究方向和解决方案。通过综合分析,本文旨在为NLP领域的研究人员和从业者提供有价值的参考。
|
4天前
|
机器学习/深度学习 存储 人工智能
深度学习在图像识别中的应用与挑战
【9月更文挑战第27天】本文将深入探讨深度学习技术如何革新了图像识别领域,并分析当前面临的主要挑战。通过简明扼要的介绍,我们将揭示深度学习模型如何超越传统方法,以及它们在实际应用中的限制和未来发展方向。
|
4天前
|
机器学习/深度学习 算法框架/工具 计算机视觉
深度学习在图像识别中的应用
【9月更文挑战第27天】本文将探讨深度学习技术如何改变图像识别领域。我们将通过实际案例和代码示例,展示深度学习模型如何从原始像素中学习和提取特征,以及如何使用这些特征进行准确的图像分类。
22 10
|
1天前
|
机器学习/深度学习 监控 算法
深度学习在图像识别中的应用与挑战
【9月更文挑战第30天】本文将探讨深度学习如何革新了图像识别领域,并介绍了一些核心算法和模型。我们将通过实际案例了解深度学习如何提升图像识别的准确度,并讨论当前面临的主要挑战及未来发展方向。
|
1天前
|
机器学习/深度学习 算法框架/工具 计算机视觉
深度学习在图像识别中的应用
【9月更文挑战第30天】本文将深入探讨深度学习技术在图像识别领域的应用。我们将首先介绍深度学习的基本原理,然后通过一个实际的代码示例,展示如何使用深度学习进行图像识别。最后,我们将讨论深度学习在图像识别中的优势和挑战。
|
3天前
|
机器学习/深度学习 存储 边缘计算
深度学习在图像识别中的应用与挑战
本文将探讨深度学习在图像识别领域中的应用及其面临的主要挑战。我们将从深度学习的基本概念入手,了解其如何改变图像识别技术。接着,通过案例分析展示深度学习在图像识别中的实际应用,并讨论当前面临的一些主要挑战,如数据隐私、算法偏见和计算资源问题。最后,我们将展望未来的发展方向和可能的解决方案。
15 4