vnr网络mimoji

作者|咸鱼

编辑| CV君

报道|我爱电脑视觉微信ID: aicvml)

论文:https://arxiv.org/abs/2103.14858

链接:https://github.com/ding3820/MIMO-VRN

编者按:在本文中,IRN视频是基于视频缩放任务进行超级分词的。与普通的VSR不同,在学习任务中加入了下采样,这可能是后VSR时代一个很好的研究方向。

看点

,最近的研究大多集中在基于图像的上下采样联合优化方案上,没有考虑时间信息。为此,本文提出了两种基于耦合层可逆神经网络的联合优化方案。

长期记忆视频缩放网络LSTM-VRN)利用低分辨率视频中的时间信息来预测丢失的高频信息。多输入多输出视频缩放网络MIMO-VRN)是一种同时缩放一组视频帧的策略。

它们不仅在定量和定性结果上优于基于图像的可逆模型,而且在视觉质量上提供了与双三次缩减视频相当的LR视频,并且在没有联合优化的情况下大大改进了视频缩放方法。这项工作是联合优化视频缩放任务的首次尝试。

方法

00-1010在介绍本文的方法之前,我们先了解一下SISR的类似方法。视频缩放任务将固定的下采样方法转化为可学习的模型,使得LR图像能够适应联合学习的上采样过程。训练目标通常要求LR图像适合人类感知。最近,IRN在这个联合优化任务中引入了一个可逆模型。它可以通过相同的神经网络结构实现图像的上下采样,为清晰地建模高斯噪声下采样丢失的高频信息提供了一种方法。

IRN正演模型包括一个二维哈尔变换和八个耦合层,如下图所示。通过二维哈尔变换,首先将输入的心率帧分解为一个低频分量和三个高频分量。然后通过耦合层处理这两个分量,以输出视觉上令人愉悦的LR图像和HR图像中固有的互补高频信息。

理论上,因为模型是可逆的,所以反向耦合层可以从Y和Z方向恢复而没有损失。但实际上,是不可能推断出来的。IRN的训练需要遵循高斯分布,这样就可以得到一个高斯样本来代替推理中缺失的高频分量。本质上,IRN是一种基于图像的方法,因此尽管IRN在图像缩放方面取得了良好的效果,但它对于视频缩放来说并不是最佳的。

IRN

LSTM-VRN和MIMO-VRN结构如下图所示。

Overview

和大多数视频超分辨率技术一样,LSTM-VRN上图a))采用SISO策略通过可逆网络的前向模型进行下采样,然后采用MISO上采样方法。

具体来说,首先,基于LSTM的预测模块融合当前LR帧的相邻帧以形成丢失高频分量的估计,该估计将被输入可逆模块以重构SR帧。

其中,需要从多个LR帧中进行推断,以决定LR视频中应该保留哪些信息,从而提高预测效果。在推理时间t,预测模块的前一模式是:

其中是sigmoid函数,是标准卷积和矩阵乘法。然后,前向传播和后向传播通过1X1卷积输出预测。注意,LSTM-VRN只使用LR视频帧的时间信息进行缩放,而其下采样仍然基于SISO方案,没有使用HR视频帧的时间信息。

LSTM-VRN

在这里,本文提出了一种新的

的基本处理单元,称为帧组Group of Frames, GoF)。首先,将HR视频分解为非重叠的帧组,…,每个帧组包含g个帧,每个帧组都独立的进行降采样。

在一个GoF中,每个视频帧首先使用二维Haar小波单独变换,得到低频分量和高频分量。然后将整个组输入到耦合层中,通过两个降尺度模块,得到一组量化的LR帧和一组高频分量。

值得注意的是,由于组耦合的性质,与之间不存在一一对应关系。 上采样的过程也是在帧组基础上进行的。如上图b)所示,本文采用基于残差块的预测模块对对应的LR帧组中缺失的高频分量进行预测。与降采样的帧组输入的概念类似,LR帧组和估计的高频分量组输入进可逆网络以逆模式运行。这种基于MIMO的上采样方法可以同时重构一组HR帧。

损失

LSTM-VRN的训练包含两个损失函数。首先,为了确保LR视频具有视觉上的美观性,将LR损失定义为:

其次,为了使HR重构质量最大化,使用Charbonnier来定义HR损失。总的损失为,其中λ为超参数。 MIMO-VRN的训练与LSTM-VRN具有相同的和损失,因为它们有共同的优化目标。

然而,我们注意到MIMO-VRN往往在GoF的视频帧上具有不均匀的HR重建质量见消融实验)。为了缓解GoF中的质量波动,在MIMO-VRN中引入了以下中心损失:

其中g是组的大小,

表示一个GoF中平均HR重构误差,M表示一个序列中GoF的个数。这个损失鼓励GoF中每个视频帧的HR重构误差近似于平均水平。

实验

中心损失的消融实验

预测模块的消融实验

Vid4上的定量评估

包含中心损失的MIMO-VRN方法成为MIMO-VRN-C,†表示采用联合优化的缩放模型。

Published by

风君子

独自遨游何稽首 揭天掀地慰生平

发表回复

您的电子邮箱地址不会被公开。 必填项已用 * 标注