代码实现——注意力引导的CNN图像去噪(Attention-guided CNN for image denoising)

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

今天给大家带来一篇注意力引导的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:重建无噪声图片的网络。该处同样采用了跳跃连接,也是为了让网络学习残差,更有利于网络学习。

更多的可以去看论文。

 

评论 32
成就一亿技术人!
拼手气红包6.0元
还能输入1000个字符
 
红包 添加红包
表情包 插入表情
 条评论被折叠 查看
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值