基于窗口自注意力网络的单图像去雨算法

高涛, 文渊博, 陈婷, 张静

A Single Image Deraining Algorithm Based on Swin Transformer

GAO Tao, WEN Yuanbo, CHEN Ting, ZHANG Jing

表2 测试数据集Test1200[12]上的网络组成消融实验对比结果

Tab.2 Comparative results of network components ablation study on Test1200[12] dataset

网络组成 组合方式 输入层 单卷积 √ √ √ √ √ × CGB × × × × × √ 特征提取网络 ResNet √ × × × × × STB × √ √ √ √ √ RSTB × × √ √ √ √ DRSTB × × × √ √ √ DRST × × × × √ √ 输出层 单卷积 √ √ √ √ √ √ PSNR/dB 25.41 27.35 28.94 30.27 32.15 34.83 SSIM 0.846 0.882 0.886 0.904 0.912 0.924