今天给大家带来一篇注意力引导的CNN图像去噪文章《Attention-guided CNN for image denoising》。我用tensorflow的实现版本可以在我的github上找到。传送门:https://github.com/sun126/ADNet。
整体的网络结构如下图:

主要分为四个部分:SBNet,FEBNet,ABNet,RBNet。
SBNet:是由CBR(Conv+BN+ReLU)与DBR(Dilated Conv+BN+ReLU)交错组成的稀疏块。Dilated Conv是空洞卷积,目的是不在池化导致信息损失的情况下增大感受野。作者在论文后面论述了此处设计的合理性。
FEBNet:利用跳跃连接融合浅层与深层的信息,提高深层网路的表达能力(ResNet思想)。
ABNet:将前面提取的结果作为权重因子,赋予图片不同位置不同的权重,目的是挖掘潜藏在复杂背景中的噪声。
RBNet:重建无噪声图片的网络。该处同样采用了跳跃连接,也是为了让网络学习残差,更有利于网络学习。
更多的可以去看论文。

本文介绍了使用注意力引导的CNN进行图像去噪的方法,包括SBNet、FEBNet、ABNet和RBNet四个部分。SBNet采用交错的CBR和DBR块,FEBNet利用跳跃连接融合信息,ABNet根据特征分配权重,RBNet进行图像重建。提供了tensorflow实现的代码仓库链接,以及训练和测试的步骤。
最低0.47元/天 解锁文章
105





