基于掩模和非局部注意力的双阶段去雨网络

侯玉珍,沈晓红,李莉,杨明源,张彩明

Dual-stage deraining network based on mask and non-local attention

Yuzhen HOU,Xiaohong SHEN,Li LI,Mingyuan YANG,Caiming ZHANG

表 2 本研究方法不同模块消融实验的结果

Tab.2 Results of ablation experiments on different modules of proposed method

模型 编码器 解码器 PSNR/dB SSIM 不分阶段 模型1 STB STB 40.69 0.9852 模型2 MAB MAB 40.99 0.9851 模型3 NAB NAB 41.10 0.9862 分阶段 模型4 MAB STB 41.41 0.9869 模型5 STB NAB 41.37 0.9871 本模型 MAB NAB 41.71 0.9907