CTNet:西工大、哈工大、中南大学和台湾清华大学一种交叉 Transformer 的图像去噪方法...

本文主要是介绍CTNet:西工大、哈工大、中南大学和台湾清华大学一种交叉 Transformer 的图像去噪方法...,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!

关注公众号,发现CV技术之美

99d0bc8795efcd591b60e4db9941ffd7.png

  • 作者:田春伟,郑梦华,左旺孟,张师超,张艳宁,林嘉文

  • 单位:西北工业大学、空天地海一体化大数据应用技术国家工程实验室、哈尔滨工业大学、中南大学、台湾国立清华大学

  • 论文题目:A cross Transformer for image denoising

  • 论文链接:https://arxiv.org/abs/2310.10408

  • 项目链接:https://github.com/hellloxiaotian/CTNet

摘要

深度卷积神经网络 (Convolutional Neural Networks, CNNs) 依赖前向传播和反向传播获得显著的图像去噪性能。但如何通过 CNNs 获取有效的结构信息解决复杂场景的图像去噪问题是关键。作者们提出了一种交叉 Transformer 去噪卷积神经网络 (Cross Transformer denoising CNN, CTNet)用于获得复杂场景下清晰图像。

本方法包括三部分来抑制噪声:串行块、并行块和残差块 。串行块使用增强的残差架构深入搜索图像的结构信息以提高图像去噪性能。并行块使用三个异构网络来实现多层特征的多重交互来防止关键信息的丢失和增强去噪模型对复杂场景的适应性。通过交互Transformer以深度和广度搜索方式增强像素特征之间关系,提高图像去噪效果。最后使用残差块来重建图像。

实验结果表明提出的 CTNet 在定性分析和定量分析上已经获得好的去噪性能,并且适用于移动数字设备。相关代码能在https://github.com/hellloxiaotian/CTNet处被获取。

主要贡献:

  1. 采用深度和广度搜索以串并行方式来获得更多的结构信息,提高图像去噪性能。

  2. 利用三个并行异构网络实现多层特征交互,增强不同网络间的关系,以提高针去噪网络对复杂场景的适应性。

  3. 交叉 Transformer 技术可以根据像素关系提取显著性特征,从而优化去噪效果。

方法

网络结构如图1所示:

35b56e7754240362f97083063a386c02.png
图1. 交叉Transformer网络结构图

实验

本文提出的方法在Set12,BSD68,Urban100,CBSD68,Kodak24,McMaster,CC,SIDD和Mayo这九个基准数据上超过了很多流行的方法,如:DnCNN、ADNet和IPT等。更多的结构如表1到表8:

0eca02672633b24727faaf0791e4ceb1.png
表1 不同图像去噪方法在BSD68数据集上对于三种不同噪声等级(15、25和50)的PSNR结果
c7b67fb4c810b49b5ec7ee8ebc89ae0d.png
表2 不同图像去噪方法在Set12数据集上对于三种不同噪声等级(15、25和50)的PSNR结果
e825e02bfd75ae2a960adcd9744cea07.png
表3 不同图像去噪方法在Urban100数据集上对于三种不同等级噪声等级(15、25、50)的PSNR结果
654c4047d3977c8501fe549a1515635b.png
表4 不同图像去噪方法在CBSD68数据集上对于五种不同等级噪声等级(15、25、35、50和75)的PSNR结果
3113777689756c3d3393181451d2f2b2.png
表5 不同图像去噪方法在Kodak24数据集上对于五种不同噪声等级(15、25、35、50和75)的PSNR结果
a168e5020ca49825e5bea8cfe2b0f708.png
表6 不同图像去噪方法在McMaster数据集上对于五种不同噪声等级(15、25、35、50和75)的PSNR结果
e24190410b0a908aa7cacae58b28e9d6.png
表7 不同图像去噪方法在CC数据集上对于真实噪声的PSNR结果
ba0acf163dff53d6c1cdf3d112306d18.png
表8 不同图像去噪方法在SIDD和Mayo数据集上对于智能手机图像和CT图像的PSNR结果

此外,本文也制作了4组可视化结果来验证本文提出的方法的有效性,如图2-图5所示:

a1360988e93e10fb6da9add111f996b5.png
图2 不同图像去噪方法在BSD68灰度图像(噪声等级为50)上的去噪效果图。(a) 原图(b) 噪声图像 (c) ADNet/27.13dB (d) DnCNN/27.92dB (e) FFDNet/28.04dB (f) CTNet/28.16dB.
cca835c3ef6126bcf3dcca024c43c683.png
图3 不同图像去噪方法在Set12灰度图像(噪声等级为25)上的去噪效果图。(a) 原图 (b)噪声图像 (c) ADNet/32.68dB (d) DnCNN/32.44dB (e) FFDNet/32.68 dB (f) CTNet/32.87dB.
1907dde9ee227fddcad2b250a8fea743.png
图4 不同图像去噪方法在CBSD68彩色图像(噪声等级为35)上的去噪效果图。(a)原图 (b) 噪声图像 (c) ADNet/28.84dB (d) DnCNN/29.33dB (e) FFDNet/29.30 dB (f) CTNet/29.90 dB.
d680c214dd5f554fa88597cd69239758.png
图5 不同图像去噪方法在McMaster彩色图像(噪声等级为50)上的去噪效果图。(a)原图 (b) 噪声图像(c) ADNet/23.70 dB (d) DnCNN/24.07 dB (e) FFDNet/25.56 dB (f) CTNet/26.39 dB.

结论

在本文中,作者们提出了一种交叉Transformer去噪卷积神经网络(CTNet)来实现图像去噪。

本方法包括三部分来抑制噪声:串行块、并行块和残差块。串行模块通过深度搜索引导增强的残差架构来获得更准确的结构信息。避免关键信息的损失,并行模块根据广度搜索的思想设计了三个并行的异构网络,实现了特征之间的多重交互,提取了更丰富的细节信息。

此外,Transformer机制被嵌入到串行模块和并行模块中,以有效地提取互补的显著性信息来去除噪声。最后,使用残差块来重建清晰的图像。

根据大量的实验分析,所提出的方法非常适合复杂场景的图像去噪。它不仅适用于移动数字设备,而且对医疗成像设备也很有用。此外,作者们将在未来的工作中研究没有参照图像的图像去噪任务。

030e53267b0c68353825b75786fb66a2.jpeg

END

欢迎加入「图像去噪交流群👇备注:去噪

dac222e7ac534ff1447e4580dcd75149.png

这篇关于CTNet:西工大、哈工大、中南大学和台湾清华大学一种交叉 Transformer 的图像去噪方法...的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!



http://www.chinasem.cn/article/381629

相关文章

Linux换行符的使用方法详解

《Linux换行符的使用方法详解》本文介绍了Linux中常用的换行符LF及其在文件中的表示,展示了如何使用sed命令替换换行符,并列举了与换行符处理相关的Linux命令,通过代码讲解的非常详细,需要的... 目录简介检测文件中的换行符使用 cat -A 查看换行符使用 od -c 检查字符换行符格式转换将

SpringBoot实现数据库读写分离的3种方法小结

《SpringBoot实现数据库读写分离的3种方法小结》为了提高系统的读写性能和可用性,读写分离是一种经典的数据库架构模式,在SpringBoot应用中,有多种方式可以实现数据库读写分离,本文将介绍三... 目录一、数据库读写分离概述二、方案一:基于AbstractRoutingDataSource实现动态

Java中的String.valueOf()和toString()方法区别小结

《Java中的String.valueOf()和toString()方法区别小结》字符串操作是开发者日常编程任务中不可或缺的一部分,转换为字符串是一种常见需求,其中最常见的就是String.value... 目录String.valueOf()方法方法定义方法实现使用示例使用场景toString()方法方法

Java中List的contains()方法的使用小结

《Java中List的contains()方法的使用小结》List的contains()方法用于检查列表中是否包含指定的元素,借助equals()方法进行判断,下面就来介绍Java中List的c... 目录详细展开1. 方法签名2. 工作原理3. 使用示例4. 注意事项总结结论:List 的 contain

macOS无效Launchpad图标轻松删除的4 种实用方法

《macOS无效Launchpad图标轻松删除的4种实用方法》mac中不在appstore上下载的应用经常在删除后它的图标还残留在launchpad中,并且长按图标也不会出现删除符号,下面解决这个问... 在 MACOS 上,Launchpad(也就是「启动台」)是一个便捷的 App 启动工具。但有时候,应

SpringBoot日志配置SLF4J和Logback的方法实现

《SpringBoot日志配置SLF4J和Logback的方法实现》日志记录是不可或缺的一部分,本文主要介绍了SpringBoot日志配置SLF4J和Logback的方法实现,文中通过示例代码介绍的非... 目录一、前言二、案例一:初识日志三、案例二:使用Lombok输出日志四、案例三:配置Logback一

Python实现无痛修改第三方库源码的方法详解

《Python实现无痛修改第三方库源码的方法详解》很多时候,我们下载的第三方库是不会有需求不满足的情况,但也有极少的情况,第三方库没有兼顾到需求,本文将介绍几个修改源码的操作,大家可以根据需求进行选择... 目录需求不符合模拟示例 1. 修改源文件2. 继承修改3. 猴子补丁4. 追踪局部变量需求不符合很

mysql出现ERROR 2003 (HY000): Can‘t connect to MySQL server on ‘localhost‘ (10061)的解决方法

《mysql出现ERROR2003(HY000):Can‘tconnecttoMySQLserveron‘localhost‘(10061)的解决方法》本文主要介绍了mysql出现... 目录前言:第一步:第二步:第三步:总结:前言:当你想通过命令窗口想打开mysql时候发现提http://www.cpp

Mysql删除几亿条数据表中的部分数据的方法实现

《Mysql删除几亿条数据表中的部分数据的方法实现》在MySQL中删除一个大表中的数据时,需要特别注意操作的性能和对系统的影响,本文主要介绍了Mysql删除几亿条数据表中的部分数据的方法实现,具有一定... 目录1、需求2、方案1. 使用 DELETE 语句分批删除2. 使用 INPLACE ALTER T

MySQL INSERT语句实现当记录不存在时插入的几种方法

《MySQLINSERT语句实现当记录不存在时插入的几种方法》MySQL的INSERT语句是用于向数据库表中插入新记录的关键命令,下面:本文主要介绍MySQLINSERT语句实现当记录不存在时... 目录使用 INSERT IGNORE使用 ON DUPLICATE KEY UPDATE使用 REPLACE