超分算法之SRCNN-程序员宅基地

技术标签: 超分  算法  计算机视觉  深度学习  人工智能  超分辨率重建  

在这里插入图片描述

这篇文章是2014年的一篇论文,其主要意义在于作者推出的SRCNN是深度学习在超分上开篇之作!SRCNN证明了深度学习在超分领域的应用可以超越传统的插值等办法取得较高的表现力。

参考目录:
深度学习图像超分辨率开山之作SRCNN(一)原理分析
深度学习端到端超分辨率方法发展历程

1 SRCNN简介

  1. 作者推出了一种基于SISR的超分方法。这种方法基于深度学习,旨在实现一种端对端的网络模型——SRCNN,其用于将低分辨率的图像转换为高分辨图像。作者指出,SRCNN在当时的数据集下达到了SOAT的水平。

  2. SRCNN具有结构简单且低失真度的特点:在这里插入图片描述如上图所示,只需要一定的训练回合,SRCNN就可以超过传统的超分方法。

  3. 在一定数量的卷积层结构下,SRCNN可以达到fast-training。

  4. 实验表明,在一定范围内,越大的数据集和较大的网络模型可以提升SRCNN对图像的重建效果。

2 SRCNN模型结构

在这里插入图片描述

首先说明以下符号的含义:

  1. Y Y Y:输入图像经过预处理(双三次插值)得到的图像,我们仍将 Y Y Y当作是低分辨率图像,但它的size要比输入图像要大。
  2. F ( Y ) F(Y) F(Y):网络最后输出的图像,我们的目标就是通过优化 F ( Y ) F(Y) F(Y)和Ground-Truth之间的loss来学会这个函数 F ( ⋅ ) F(\cdot) F()
  3. X X X:高分辨率图像,即Ground-Truth,它和 Y Y Y的size是相同的。

如上图所示是SRCNN的网络模型,其分为三部分,分别是:
①:Patch extraction and representation(其实就是图像特征提取层)。通过CNN将图像 Y Y Y的特征提取出来存到向量中,这个向量里包含了多张feature map,即一张图所含的一些特征。
②:非线性映射层。将上一层的feature map进一步做非线性映射处理,使得网络深度加大,更有利于学到东西。
③:网络重建层。重建用于将feature map进行还原成高分辨率图像 F ( Y ) F(Y) F(Y),其与 X X X做loss并通过反传来学习整个模型的参数。

下面分别详细展开论述上面三个层。

特征提取层:
特征提取层用了一层的CNN以及ReLU去将图像 Y Y Y变成一堆堆向量,即feature map:
F 1 ( Y ) = m a x ( 0 , W 1 ⋅ Y + B 1 ) . F_1(Y) = max(0, W_1\cdot Y+B_1). F1(Y)=max(0,W1Y+B1).其中 W 1 、 B 1 W_1、B_1 W1B1是滤波器(卷积核)的参数,这是一个 f 1 × f 1 f_1\times f_1 f1×f1大小的窗口,通道数为 Y Y Y的通道 c c c,一共有 n 1 n_1 n1个滤波器。
Note:

  1. 经过这一层,图像 Y Y Y的大小以及通道数都会发生改变。
  2. m a x ( 0 , x ) max(0,x) max(0,x)表示ReLU层。

非线性映射层:
这一层就是将上一层的feature map再用卷积核过滤一次以及ReLU层进行激活,也可以理解为为了加深网络从而更好的学习函数 F ( ⋅ ) F(\cdot) F()
F 2 ( Y ) = m a x ( 0 , W 2 ⋅ F 1 ( X ) + B 2 ) . F_2(Y) = max(0, W_2\cdot F_1(X)+B_2). F2(Y)=max(0,W2F1(X)+B2).大致结构和特征提取层一样,不一样的是这一层只是为了增加网络模型的非线性程度,所以只需采用 1 × 1 1\times 1 1×1的卷积核就可以了,其通道数为 n 1 n_1 n1,一共有 n 2 n_2 n2个滤波器。当然可以继续增加非线性层,但是本文旨在推出一种通用性SR框架,所以会选择最简的网络模型

图像重建层:
借鉴于传统超分的纯插值办法——对图像局部进行平均化的思想,其本质就是乘加结合的方式,因此作者决定采用卷积的方式(也是乘加结合的方式)去做重建:
F ( Y ) = W 3 ⋅ F 2 ( Y ) + B 3 . F(Y) = W_3\cdot F_2(Y) + B_3. F(Y)=W3F2(Y)+B3.这一层是不需要ReLU层的,且卷积核的大小为 n 2 × c × f 3 × f 3 n_2\times c \times f_3 \times f_3 n2×c×f3×f3.
Note:

  1. 也可以从另一个角度来考虑,经过前面的卷积之后,图像的size变小了,因此需要上采样过程来恢复图像,势必需要一个反卷积来做这件事,而反卷积本质也是卷积的一种。

3 Loss function:

设batchsize为 n n n,SRCNN网络参数集为 Θ = { W 1 , W 2 , W 3 , B 1 , B 2 , B 3 } \Theta = \{W_1, W_2, W_3, B_1, B_2, B_3\} Θ={ W1,W2,W3,B1,B2,B3},则Loss function可定义为:
L ( Θ ) = 1 n ∑ i = 1 n ∣ ∣ F ( Y i ; Θ ) − X i ∣ ∣ 2 . L(\Theta) = \frac{1}{n}\sum^n_{i=1}||F(Y_i;\Theta) - X_i||^2. L(Θ)=n1i=1nF(Yi;Θ)Xi2.Note:

  1. 选择MSE作为损失函数的一个重要原因是MSE的格式和我们图像失真评价指标PSNR很像,因此可以理解为SRCNN是直接冲着提升PSNR去的,从而让高分辨率的图像有较小的失真度。
  2. MSE就是迫使网络将我们恢复的SR图像向着Ground-Truth(标签 X X X)的方向靠近。

4 实验

4.1 setup

实验的一些比较重要的配置如下:

  1. Training-data涉及91张图片。Set5数据集涉及5张图片用于up-scale-factor={2,3,4}的验证与测试;Set14数据集涉及14张图片用于up-scale-factor=3的验证与测试。
  2. 实验的一些参数设置: f 1 = 9 , f 3 = 5 , n 1 = 64 , n 2 = 32 f_1=9,f_3=5,n_1=64,n_2=32 f1=9,f3=5,n1=64,n2=32.
  3. Ground-Truth的大小是 32 × 32 32\times 32 32×32
  4. 卷积核的参数初始化来自于: w i 0 ∼ N ( 0 , 0.001 ) w_i^0\sim\mathcal{N}(0, 0.001) wi0N(0,0.001)
  5. SRCNN一共3层网络,前两层配置的学习率为 1 0 − 4 10^{-4} 104,最后一层的学习率配置为 1 0 − 5 10^{-5} 105,作者指出这种让最后一层较小的学习率有利于网络收敛。

4.2 实验结果

4.2.1 performance

在这里插入图片描述从上图看出SRCNN的PSNR在大部分图片中都取得了最佳的值!此外,所消耗的时间也是最少的。

4.2.2 runtime

在这里插入图片描述从上图可以看出SRCNN有最少的runtime!

5 进一步研究

5.1 滤波器学习情况

在这里插入图片描述上图是特征提取层滤波器的学习可视化图,在91张图片的训练结果,其中up-scale-factor=2。
图像a、f:类似于高斯分布。
图像b、c、d:类似于边缘检测。
图像e:类似于纹理检测。
其余:一些坏死的卷积核参数。

5.2 ImageNet学习

作者这一节旨在探究数据集的大小对performance的影响。

  1. 采用ILSVRC 2013的ImageNet数据集和91张图片这两个训练集做对比训练。
  2. 在Set5数据上做测试,up-scale-factor=3。

实验结果如下:
在这里插入图片描述从图中可知,大的数据集对表现力的提升是有帮助的(虽然我们都知道,但是作者还是做个实验来证明下)。

5.3 滤波器数量

作者研究滤波器数量对PSNR的提升影响,设置了3组实验,结果如下:
在这里插入图片描述
实验结果表明卷积核的数量对表现力是有提升的,但是数量的增加也带来了runtime的增加,如果你想获取快速的重建效果,建议还是取小数量的卷积核更好。

5.4 滤波器大小

作者研究滤波器size对PSNR的提升影响,进行了2组实验,分别是:
f 1 = 9 , f 3 = 5 f_1=9,f_3=5 f1=9,f3=5 f 1 = 11 , f 3 = 7 f_1=11,f_3=7 f1=11,f3=7
从实验结果来看,较大的卷积核可以提取更好的特征信息,但是也带来了runtime的上升,因此实际中我们需要根据实际情况进行trade-off。

6 效果展示

在这里插入图片描述

7 总结

  1. 本文作为SR在深度学习领域的开篇之作,提出了一种通用性框架SRCNN,将输入图像进行Bicubic插值预处理,然后特征提取,非线性映射,最后进行重建;重建后的图像与Ground-Truth做loss来迫使网络学习到如何从 L R → H R LR \to HR LRHR的知识。
  2. 选用深度学习常用的MSE作为Loss function,因为MSE与PSNR有着相似的表达式。
  3. SRCNN在PSNR和runtime上都表现不俗,超越了当时的SOAT,表征了这种框架的实用性。
  4. 作者做了一系列实验,其中包括可视化乐特征提取到的向量是怎么样的;大的数据集对表现力的提升是有帮助的;卷积核的数量的增加对表现力是有提升的,但是数量的增加也带来了runtime的增加;较大的卷积核可以提取更好的特征信息,但是也带来了runtime的上升。
版权声明:本文为博主原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。
本文链接:https://blog.csdn.net/MR_kdcon/article/details/123456507

智能推荐

linux devkmem 源码,linux dev/mem dev/kmem实现访问物理/虚拟内存-程序员宅基地

文章浏览阅读451次。dev/mem: 物理内存的全镜像。可以用来访问物理内存。/dev/kmem: kernel看到的虚拟内存的全镜像。可以用来访问kernel的内容。调试嵌入式Linux内核时,可能需要查看某个内核变量的值。/dev/kmem正好提供了访问内核虚拟内存的途径。现在的内核大都默认禁用了/dev/kmem,打开的方法是在 make menuconfig中选中 device drivers --> ..._dev/mem 源码实现

vxe-table 小众但功能齐全的vue表格组件-程序员宅基地

文章浏览阅读7.1k次,点赞2次,收藏19次。vxe-table,一个小众但功能齐全并支持excel操作的vue表格组件_vxe-table

(开发)bable - es6转码-程序员宅基地

文章浏览阅读62次。参考:http://www.ruanyifeng.com/blog/2016/01/babel.htmlBabelBabel是一个广泛使用的转码器,可以将ES6代码转为ES5代码,从而在现有环境执行// 转码前input.map(item => item + 1);// 转码后input.map(function (item) { return item..._让开发环境支持bable

FPGA 视频处理 FIFO 的典型应用_fpga 频分复用 视频-程序员宅基地

文章浏览阅读2.8k次,点赞6次,收藏29次。摘要:FPGA视频处理FIFO的典型应用,视频输入FIFO的作用,视频输出FIFO的作用,视频数据跨时钟域FIFO,视频缩放FIFO的作用_fpga 频分复用 视频

R语言:设置工作路径为当前文件存储路径_r语言设置工作目录到目标文件夹-程序员宅基地

文章浏览阅读575次。【代码】R语言:设置工作路径为当前文件存储路径。_r语言设置工作目录到目标文件夹

background 线性渐变-程序员宅基地

文章浏览阅读452次。格式:background: linear-gradient(direction, color-stop1, color-stop2, ...);<linear-gradient> = linear-gradient([ [ <angle> | to <side-or-corner>] ,]? &l..._background线性渐变

随便推点

【蓝桥杯省赛真题39】python输出最大的数 中小学青少年组蓝桥杯比赛 算法思维python编程省赛真题解析-程序员宅基地

文章浏览阅读1k次,点赞26次,收藏8次。第十三届蓝桥杯青少年组python编程省赛真题一、题目要求(注:input()输入函数的括号中不允许添加任何信息)1、编程实现给定一个正整数N,输出正整数N中各数位最大的那个数字。例如:N=132,则输出3。2、输入输出输入描述:只有一行,输入一个正整数N输出描述:只有一行,输出正整数N中各数位最大的那个数字输入样例:

网络协议的三要素-程序员宅基地

文章浏览阅读2.2k次。一个网络协议主要由以下三个要素组成:1.语法数据与控制信息的结构或格式,包括数据的组织方式、编码方式、信号电平的表示方式等。2.语义即需要发出何种控制信息,完成何种动作,以及做出何种应答,以实现数据交换的协调和差错处理。3.时序即事件实现顺序的详细说明,以实现速率匹配和排序。不完整理解:语法表示长什么样,语义表示能干什么,时序表示排序。转载于:https://blog.51cto.com/98..._网络协议三要素csdn

The Log: What every software engineer should know about real-time data's unifying abstraction-程序员宅基地

文章浏览阅读153次。主要的思想,将所有的系统都可以看作两部分,真正的数据log系统和各种各样的query engine所有的一致性由log系统来保证,其他各种query engine不需要考虑一致性,安全性,只需要不停的从log系统来同步数据,如果数据丢失或crash可以从log系统replay来恢复可以看出kafka系统在linkedin中的重要地位,不光是d..._the log: what every software engineer should know about real-time data's uni

《伟大是熬出来的》冯仑与年轻人闲话人生之一-程序员宅基地

文章浏览阅读746次。伟大是熬出来的  目录  前言  引言 时间熬成伟大:领导者要像狼一样坚忍   第一章 内圣外王——领导者的心态修炼  1. 天纵英才的自信心  2. 上天揽月的企图心  3. 誓不回头的决心  4. 宠辱不惊的平常心  5. 换位思考的同理心  6. 激情四射的热心  第二章 日清日高——领导者的高效能修炼  7. 积极主动,想到做到  8. 合理掌控自己的时间和生命  9. 制定目标,马..._当狼拖着受伤的右腿逃生时,右腿会成为前进的阻碍,它会毫不犹豫撕咬断自己的腿, 以

有源光缆AOC知识百科汇总-程序员宅基地

文章浏览阅读285次。在当今的大数据时代,人们对高速度和高带宽的需求越来越大,迫切希望有一种新型产品来作为高性能计算和数据中心的主要传输媒质,所以有源光缆(AOC)在这种环境下诞生了。有源光缆究竟是什么呢?应用在哪些领域,有什么优势呢?易天将为您解答!有源光缆(Active Optical Cables,简称AOC)是两端装有光收发器件的光纤线缆,主要构成部件分为光路和电路两部分。作为一种高性能计..._aoc 光缆

浏览器代理服务器自动配置脚本设置方法-程序员宅基地

文章浏览阅读2.2k次。在“桌面”上按快捷键“Ctrl+R”,调出“运行”窗口。接着,在“打开”后的输入框中输入“Gpedit.msc”。并按“确定”按钮。如下图 找到“用户配置”下的“Windows设置”下的“Internet Explorer 维护”的“连接”,双击选择“自动浏览器配置”。如下图 选择“自动启动配置”,并在下面的“自动代理URL”中填写相应的PAC文件地址。如下..._設置proxy腳本