融合3D注意力和Transformer的图像去雨网络

文档名：融合3D注意力和Transformer的图像去雨网络
目的：因为有雨图像中雨线存在方向、密度和大小等各方面的差异,单幅图像去雨依旧是一个充满挑战的研究问题.现有算法在某些复杂图像上仍存在过度去雨或去雨不足等问题,部分复杂图像的边缘高频信息在去雨过程中被抹除,或图像中残留雨成分.针对上述问题,本文提出三维注意力和Transformer去雨网络(three-dimensionattentionandTransformerderainingnetwork,TDATDN).方法：将三维注意力机制与残差密集块结构相结合,以解决残差密集块通道高维度特征融合问题;使用Transformer计算特征全局关联性;针对去雨过程中图像高频信息被破坏和结构信息被抹除的问题,将多尺度结构相似性损失与常用图像去雨损失函数结合参与去雨网络训练.结果：本文将提出的TDATDN网络在Rain12000雨线数据集上进行实验.其中,峰值信噪比(peaksignaltonoiseratio,PSNR)达到33.01dB,结构相似性(structuralsimilarity,SSIM)达到0.9278.实验结果表明,本文算法对比以往基于深度学习的神经网络去雨算法,显著改善了单幅图像去雨效果.结论：本文提出的TDATDN图像去雨网络结合了3D注意力机制、Transformer和编码器—解码器架构的优点,可较好地完成单幅图像去雨工作.
作者：王美华 柯凡晖 梁云 范衠 廖磊
作者单位：华南农业大学数学与信息学院,广州510642汕头大学工学院,汕头515063
母体文献：2022低质图像增强前沿论坛论文集
会议名称：2022低质图像增强前沿论坛  
会议时间：2022年4月21日
会议地点：线上
主办单位：中国图象图形学报
语种：chi
分类号：
关键词：图像去雨  特征融合  3D注意力  Transformer去雨网络
在线出版日期：2022年9月21日
基金项目：
相似文献
相关博文

融合3D注意力和Transformer的图像去雨网络.pdf

文件大小:

22.44 MB

高速下载