基于深度卷积神经网络和跳跃连接的图像去噪和超分辨

Image Restoration Using Very Deep Convolutional Encoder-Decoder Networks with Symmetric Skip Connections

作者:Xiao-Jiao Mao、Chunhua Shen等

本文提出了一个深度的全卷积编码-解码框架来解决去噪和超分辨之类的图像修复问题。网络由多层的卷积和反卷积组成,学习一个从受损图像到原始图像的端到端的映射。卷积层负责特征提取,捕获图像内容的抽象信息,同时消除噪声/损失。相对应,反卷积层用来恢复图像细节。

基于深度卷积神经网络和跳跃连接的图像去噪和超分辨

网络结构

基于深度卷积神经网络和跳跃连接的图像去噪和超分辨

在底层图像修复领域,由于池化操作会丢失有用的图像细节信息,因此,本网络没有用到池化层,是一个全卷积的网络结构。

卷积层的特征图和与之相应成镜像关系的反卷积层特征图进行跳跃连接,对应像素直接相加后经过非线性激活层然后传入下一层。

受 VGG 模型的启发,卷积核大小都设置为为 3*3。另外,由于网络结构本质上是一个像素级的预测,因此输入可以是任意大小的图片,输出和输入保持一致。

基于深度卷积神经网络和跳跃连接的图像去噪和超分辨

反卷积解码器

在全卷积的情况下,噪声一步步地被消除。经过每一个卷积层后,噪声等级减小,图像内容的细节也可能会随之丢失。在本篇论文提出的结构中,卷积层保留了主要的图像内容,而反卷积层则用来补偿细节信息,可以达到良好去噪效果的同时较好地保留图像内容。

另一方面,卷积层逐渐减小特征图的大小,反卷积层再逐渐增大特征图的大小,最终确保输入输出大小一致,也可以保证在移动端计算能力有限情况下的测试效率。

基于深度卷积神经网络和跳跃连接的图像去噪和超分辨

跳跃连接

正如残差网络的设计初衷,跳跃连接可以解决网络层数较深的情况下梯度消失的问题,同时有助于梯度的反向传播,加快训练过程。

基于深度卷积神经网络和跳跃连接的图像去噪和超分辨

通过传递卷积层的特征图到反卷积层,有助于解码器拥有更多图像细节信息,从而恢复出更好的干净图像。

针对输入噪声图像 X 和输出干净图像 Y,本网络致力于学习一个残差即 F(X) = Y - X。

基于深度卷积神经网络和跳跃连接的图像去噪和超分辨

讨论

基于深度卷积神经网络和跳跃连接的图像去噪和超分辨

实验 (a) 对比有无跳跃连接情况下 PSNR 随迭代次数的变化,可以看到有跳跃连接的情况下 PSNR 有显著提高。

实验 (b) 对比不同网络结构下 Loss 值随迭代次数的变化,可以看到在同等层数的网络下,有跳跃连接时 Loss 值相对要小很多。

实验 (c) 对比一般的残差网络(由一系列的残差块组成)和本文提出的网络情况下 PSNR 随迭代次数的变化,可以看到本文中提出的这种跳跃连接方式可以获得更好的 PSNR。

网络的大容量使得本文提出的结构能够处理不同噪声等级的图像去噪问题和不同尺寸参数的超分辨问题。

基于深度卷积神经网络和跳跃连接的图像去噪和超分辨

实验

因为本网络的卷积核仅仅是为了消除噪声,因此对图像内容的方向不敏感。所以,在测试的时候,我们可以旋转或者镜像翻转卷积核进行多次前向传播,然后对多次的输出取平均从而得到一个更平滑的结果。

图像去噪的实验结果对比

基于深度卷积神经网络和跳跃连接的图像去噪和超分辨

图像超分辨的实验结果对比

基于深度卷积神经网络和跳跃连接的图像去噪和超分辨

可以看到,RED-20, 10 层卷积 10 层反卷积的情况下已经取得了比传统方法好的效果,而且加深网络之后还可以取得更好的效果。

噪声等级越大,本文中提出的网络比其他网络的提高幅度就越大,也就是优势越明显。

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 194,491评论 5 459
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 81,856评论 2 371
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 141,745评论 0 319
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 52,196评论 1 263
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 61,073评论 4 355
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 46,112评论 1 272
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 36,531评论 3 381
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 35,215评论 0 253
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 39,485评论 1 290
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 34,578评论 2 309
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 36,356评论 1 326
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 32,215评论 3 312
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 37,583评论 3 299
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 28,898评论 0 17
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 30,174评论 1 250
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 41,497评论 2 341
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 40,697评论 2 335

推荐阅读更多精彩内容