基于掩模和非局部注意力的双阶段去雨网络
侯玉珍,沈晓红,李莉,杨明源,张彩明
Dual-stage deraining network based on mask and non-local attention
Yuzhen HOU,Xiaohong SHEN,Li LI,Mingyuan YANG,Caiming ZHANG
图 4
非局部注意力模块(NAB)结构图
Fig.4
Structure of non-local attention block(NAB)