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

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

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

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

图 2 雨纹掩模生成模块

Fig.2 Rain streak mask generation block