面向多天气退化图像恢复的自注意力扩散模型

秦菁, 文渊博, 高涛, 刘瑶

A Transformer-Based Diffusion Model for All-in-One Weather-Degraded Image Restoration

QIN Jing, WEN Yuanbo, GAO Tao, LIU Yao

表7 噪声估计网络的组件消融实验结果

Tab.7 Ablation experimental results of individual components of noise estimation network

网络模型 网络组成 单步估计 用时/s Raindrop-A ResBlock ViT SA TSA STSA FFN SGFFN DGGFFN PSNR / SSIM 基线网络 √ √ 0.6634 29.17/0.9162 ⅰ √ √ √ 1.0725 29.86/0.9203 ⅱ √ √ √ 0.8603 31.54/0.9297 ⅲ √ √ √ 0.3961 30.49/0.9381 ⅳ √ √ √ 0.4033 31.73 _/ 0.9394 _ AWIR-TDM √ √ √ 0.4080 _ 32.33/0.9429