Deep Video Super-Resolution Network Using Dynamic Upsampling Filters Without Explicit Motion Comp...

题目

Deep Video Super-Resolution Network Using Dynamic Upsampling Filters Without Explicit Motion Compensation

摘要

近年来,视频超分辨率(VSR)是对于为超高清显示提供高分辨率(HR)内容的越来越重要。但是已经提出的许多基于深度学习的VSR方法,其中大部分都严重依赖于运动估计和补偿的精度。在本文中,我们为VSR介绍了一个完全不同的框架。我们提出了一种新的端到端深度神经网络,用于生成动态上采样滤波器和残差图像,这些滤波器和残差图像是根据每个像素的局部时空邻域来计算的,以避免显式运动补偿。该方法使用动态上采样滤波器对输入图像进行直接重构,并通过计算残差增加细节。我们的网络借助一种新的数据增强技术,可以生成具有时间一致性的更清晰的HR视频。我们还通过大量的实验对我们的网络进行分析,以显示网络如何隐式地处理运动。

介绍

超分辨率(SR)的目标是从相应的低分辨率(LR)图像或视频中生成高分辨率(HR)图像或视频。SR广泛用于从医学成像到卫星成像和监视的许多领域。随着显示技术的进步,LR视频的视频超分辨率(VSR)作为超高清晰度电视的4K(3840×2160)和8K(7680×4320)目标显得越来越重要,但与之相匹配的内容仍然很少。
随着计算机视觉中的深度学习在图像分类和图像分割中的成功,出现了基于深度学习的单图像超分辨率(SISR)方法。这些方法在峰值信噪比(PSNR)和结构相似性指数(SSIM)方面显示了最新的性能。
执行VSR的一种简单方法是逐帧运行SISR。然而,由于SISR方法没有考虑帧之间的时间关系,因此很有可能连续的帧并不是自然连接的,从而导致闪烁的伪影。
传统的VSR(或多图像超分辨率)算法通过考虑相邻LR帧之间的亚像素运动输入多个LR帧和输出HR帧。所有基于深层学习的VSR方法都遵循相似的步骤,并且由两个步骤组成:运动估计和补偿过程,然后是上采样过程。这种两步法的一个问题是结果严重依赖于精确的运动估计。这种类型的方法的另一个潜在问题是HR输出帧是产生的。
这种方式的另一个潜在问题是,HR输出帧是通过卷积神经网络(CNN)混合来自多重运动补偿输入LR帧的值而产生的,这可导致输出HR帧的模糊。
在本文中,我们提出了一种新型的端到端深层神经网络,它与以往的方法有着本质的不同。运动信息被隐式地用于生成动态上采样滤波器,而不是对输入帧之间的运动进行显式计算和补偿。利用生成的上采样滤波器,HR帧直接通过到输入中心帧的局部滤波来构造(图2)。由于我们不依赖于运动的显式计算,也不直接组合来自多重帧的值,因此我们可以生成更清晰和时间一致的HR视频。


图2 Dynamic upsampling.png

使用大量的训练视频和一个新的数据增强过程,我们实现了与以往的基于深度学习的VSR算法相比的最新性能。图1显示了一个示例,与最先进的方法之一VSRnet[16]相比,我们的方法产生更锐利的帧且更少的闪烁。

相关工作

方法

VSR的目标是从给定的LR帧 {Xt} 估计HR帧 {Ŷt} 。LR帧{Xt} 从相应的GT帧 {Yt} 下采样,其中t表示时间步长。提出的VSR网络G和网络参数θ,VSR的问题被定义为
{Ŷt}=G_\theta(X_{t-N:t+N})
N是时间半径。G的一个输入张量的形状是T\times H \times W \times C,其中T = 2N + 1,H 和W是输入LR帧的高和宽,C是色彩通道的数量。对应输出张量形状是1\times rH \times rW \times C,其中 r是放大因子。
如图3所示,我们的网络从一组输入LR帧 { X_{t-N:t+N}} 产生两个输出生成最终的HR 帧Ŷt:动态upsampling过滤器F_t和残差R_t。输入中心帧X_t首先经过动态上采样upsampling过滤器F_t的局部滤波,然后残差R_t添加为最终的输出结果Ŷ_t添加上采样的upsampledỸ_t

图3 Network architecture.png

动态上采样滤波器 Dynamic Upsampling Filters

传统的双线性bilinear或双三次bicubic上采样的滤波核基本上是固定的,唯一的变化是滤波核根据上采样图像中新创建的像素的位置而移动。对于 X4 upsampling,传统的upsampling过程使用一系列16个固定的滤波核kernel。它们速度快,但很难恢复锐化和纹理区域。
与此相反,我们受动态滤波器网络 (DFN) 的灵感启发,提出使用动态上采样滤波器。上采样滤波器是根据LR帧中每个像素的时空邻域进行局部动态生成的。
VSR动态上采样过程的概述如上图2所示。首先,将一组输入LR帧{ X_{t-N:t+N}} (在我们网络中的7帧:N = 3) 送入动态滤波器生成网络。训练网络输出一组r^2 HW 确定大小(在我们的网络是5\times5)的上采样upsampling过滤器F_t,将用于生成经过滤波器的HR 帧的Ỹt_1新的像素。最后,对输入帧X_t中的LR像素进行局部滤波,得到每个输出HR像素值,对应的滤波器F_t^{y,x,v,u}如下:
\tilde{Y}_t(yr+v,xr+u)=\sum_{j=-2}^2\sum_{i=-2}^2F_t^{y,x,v,u}(j+2,i+2)X_t(y + j, x + i)
其中,x,y是LR是栅格的坐标,v和u是每个 r\times r 输出块的坐标 (0 ≤ v, u ≤r − 1),注意,这种操作类似于反卷积(或转置卷积),因此我们的网络可以进行端到端训练,因为它允许反向传播。
我们的方法与以往的基于深度学习的SR方法有本质的区别,深度神经网络学习在特征空间中通过一系列的卷积来重构HR帧。相反,我们使用深层神经网络来学习最好的上采样滤波器,然后用它直接从给定的LR帧重建HR帧。从概念上讲,动态滤波器是根据像素的运动来创建的,因为这些滤波器是通过观察像素的时空邻域来生成的,这使我们能够避免显式的运动补偿。

残差学习 Residual Learning

单独应用动态上采样滤波器后的结果缺乏清晰度(图像锐利的细节),因为它仍然是输入像素的加权和。通过线性滤波可能有不可恢复的细节。为了解决这个问题,我们另外预测残留图像以增加高频细节。在[17 ]中,将残差添加到双三次上采样基准以产生最终输出。我们的方法不同与[17]不同,因为残差图像是由多个输入帧而不是单个输入帧构成的,并且我们使用动态上采样帧作为更好的基准,然后加上计算的残差。通过组合这些互补组件,我们能够在得到的HR帧中实现空间清晰度和时间一致性。
[17] J. Kim, J. Kwon Lee, and K. Mu Lee. Accurate image super-
resolution using very deep convolutional networks. In CVPR,
pages 1646–1654, 2016.

网络设计 Network Design

如图3所示,我们的滤波器和残差生成网络被设计成共享大部分权重,因此我们可以减少产生两个不同输出所带来的开销。我们网络的共享部分是受稠密块dense block [10]的启发而设计的,并且它被适当地修改以解决我们的问题。特别地,我们用3D卷积层代替2D卷积层,以从视频数据中学习时空特征。已知3D卷积层比2D卷积层更适合于人类动作识别和视频数据的一般时空特征提取。稠密块dense block的每个部分依次由批量归一化(BN)[12]、ReLU[5]、1×1×1卷积、BN、ReLU和3×3×3卷积组成。正如在[10 ]中所做的那样,每个部分都把前面所有的特征映射作为输入。

[5] X. Glorot, A. Bordes, and Y. Bengio. Deep sparse rectifier neural networks. In AISTATS, 2011.
[10] G. Huang, Z. Liu, L. van der Maaten, and K. Q. Weinberger. Densely connected convolutional networks. In CVPR, 2017.
[12] S. Ioffe and C. Szegedy. Batch normalization: Accelerating deep network training by reducing internal covariate shift. In ICML, 2015.

每个输入LR帧首先由一个共享的二维卷积层处理,并沿时间轴连接。产生的时空特征图经过我们的3D 稠密块dense block,然后在由几个2D卷积层组成的独立分支上进行处理,生成两个输出。过滤输出\tilde{Y}_t添加生成的残余R_t,产生最终的输出\hat{Y}_t

时间累加 Temporal Augmentation

为了使提出的网络能够充分理解各种复杂的真实世界运动,我们需要相应的训练数据。为了创建这样的训练数据,我们在一般数据扩充(如随机旋转和翻转)之上,在时间轴上采用数据扩充。这里,我们引入变量T A,它决定了时间增强的采样间隔。例如,T a=2,我们将对每间隔一帧进行采样,以模拟更快的运动。当我们将T A值设置为负值时,我们还可以按照相反的顺序创建一个新的视频样本(图4)。使用各种大小的T A(在我们的工作中从-3到3),我们可以创建具有丰富移动性的训练数据。注意,在|T A|>3的情况下,VSR性能会随着对象的位移太大而降低。


图4 Create a new video sample.png

实施 Implementation

实验结果 Experimental Results

学习运动可视化 Visualization of Learned Motion

综合运动试验 为了验证提出的方法在不进行显式运动补偿的情况下,能够很好地合理利用给定视频的时域信息,合成在GT帧中每帧垂直条左右移动1个像素的视频。对于同一个输入块patch,因为每个条有不同的运动,两个视频的激活映射应该是不同的。但是生成的HR帧是相同的,所以上采样过滤器应该是相同的。在图5中(a),我们把激活第一个3×3×3的地图卷积层和生成的动态过滤器为不同的两个区域。根据预测,在相同的输入块中,我们的学习网络对不同的运动表现出不同的激活,生成的动态滤波器的权值几乎相同。
我们还在图5(b)中绘制了带点的橙色水平扫描线随时间的过渡图,以查看时间一致性是否得到很好的保护。由于VDSR[17]不考虑时间信息,所以结果是不稳定的。VSRnet[16]的重构条也呈现锯齿状伪影,在播放视频时都会产生严重的闪烁伪影。相比之下,我们的结果看起来更清晰,显示了维持时间一致性的能力,并且更接近于真值ground truth。我们建议读者阅读我们的补充视频,以便更好地进行比较。

图5.png

学习过滤器 查看学习过的过滤器,是另一种检查我们的方法能否学习和使用时间信息的方法。为此,本方法在3×3×3卷积层中的学习滤波器应该从前后以及输入到时间轴的中心提取一些值。我们可以看到,由于滤波器在时间步长t_1和t+1以及图6所示的t上具有更高的权重,因此我们的网络能够以各种结构从前后以及输入到时间轴的中心均匀地提取一些值。这表明我们的方法可以隐式地学习和使用输入帧之间的时间信息来生成精确的输出帧。
图6.png

动态上采样过滤器 生成的上采样滤波器应该针对不同的区域具有不同的权重,以自适应地处理局部运动。图7详细显示了场景日历的两个不同区域的建议的上采样过程的示例,并且每个新创建的像素位置的滤波器权重是不同的。对于具有不同纹理和运动方向的两个输入块,我们的方法自适应地生成上采样滤波器用于精确的输出重建。
图7.png

与其他方法的比较

通过我们由16层(Ours-16L)组成的基本网络,还测试了28层(Ours-28L)和52层(Ours-52L)的网络。和大多数超分辨率方法一样,具有细线条的场景非常具有挑战性,如图8所示。使用我们的算法,增加层数为这种类型的挑战性场景提供更好的结果。


图8.png

定量评价 与其他最先进的VSR方法的定量比较显示在表1中。结果表明,随着网络深度的增加,网络性能有改善的趋势,在附加参数为0.2M的情况下,Ours-28L的PSNR值比Ours-16L提高了0.18dB。即使叠加52层,我们的方法也能很好地工作,并且Vid4的PSNR值提高到27.34dB,比我们的-16L高0.53dB。即使使用我们的-16L,我们在PSNR和SSIM方面对所有高档生产商都比其他所有方法表现得好。例如,OURS16L的PSNR比第二最高结果(34)(r=4)高0.8dB。
定性比较 在图9中示出了一些定性示例。使用我们的方法更好地重建精细的细节和纹理。在图10中,我们还将[34]的结果与使用不同深度的网络的结果进行比较。我们的性能优于前面的工作,并且还可以更深入地观察性能的提高。在图11中示出了与VID4上其他最先进的VSR方法的更定性的比较。我们的研究结果表明,与其他作品相比,输出更流畅,时间更平稳。在X-T图像中锋利而平滑的边缘表明视频的抖动要小得多。

[34] X. Tao, H. Gao, R. Liao, J. Wang, and J. Jia. Detail-revealing deep video super-resolution. In ICCV, 2017.

图9.png
图10.png

结论

本文提出了一种新的基于深度学习的VSR框架,用于同时学习输出动态上采样滤波器和残差。我们用新的框架实现了最先进的性能,恢复了清晰的HR帧,并保持了时间上的一致性。通过实验,我们证明了我们的深层网络可以隐式地处理运动,而不需要显式的运动估计和补偿。
使用NVidia GeForce GTX 1080Ti,对Ours-16L、Ours-28L和Ours-52L分别进行0.4030s、0.8382s和2.8189s的7个输入帧的480×270,生成单个1920×1080输出帧需要训练2天。在我们的推理过程中,大约一半的运行时间花在局部过滤上。将来,我们将致力于加速实现实时性能的方法。我们还想扩展我们的工作,以增加时间分辨率除了空间分辨率,例如,创建一个60fps超高清视频从30fps的SD视频。

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

推荐阅读更多精彩内容