RIFE-实时视频插帧算法 | RIFE:Real-Time Intermediate Flow Estimation for Video Frame Interpolation

本文主要是介绍RIFE-实时视频插帧算法 | RIFE:Real-Time Intermediate Flow Estimation for Video Frame Interpolation,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!

关注一下实时视频插帧算法
论文地址:https://arxiv.org/pdf/2011.06294.pdf
Github地址:https://github.com/hzwer/arXiv2020-RIFE

在这里插入图片描述

Abstract:

我们为视频帧插值(VFI)提出了一种实时中间流估计算法RIFE (Real-Time Intermediate Flow Estimation)。 现有的大多数方法首先估计双向光流,然后将它们线性组合以近似中间流,从而导致运动边界周围出现伪影。 我们设计了一个中间流模型IFNet,它可以直接估计从粗到细的中间流。 然后,我们根据估计的中间流对输入帧进行warp,并采用融合过程来计算最终结果。 基于我们提出的leakage蒸馏技术,RIFE可以进行端到端的训练并获得出色的性能。 实验表明,RIFE比现有的基于流的VFI方法要快得多,并且可以在多个基准上达到最新的指标。

Introduction:

视频帧插值(VFI)

旨在合成视频的两个连续帧之间的中间帧,并广泛用于提高帧速率和增强视觉质量。VFI还支持各种应用,例如慢动作生成,视频压缩和用于视频运动去模糊的训练数据生成。 此外,以实时速度运行在高分辨率视频(例如720p和1080p)上的VFI算法具有更多潜在的应用,例如在客户端播放器上播放更高帧频的视频,为有限的用户提供视频编辑服务 计算资源。

目前VFI的常用方法包括两个步骤

1)根据近似的光流对输入帧进行warping;

2)使用卷积神经网络(CNN)融合和完善warped的帧。
在这里插入图片描述
依据输入帧如何warping,基于流的VFI算法可以分为基于forward warping的方法和基于backward warping的方法。目前主要采用后者的warping方式。给定输入帧 I 0 I_{0} I0 I 1 I_{1} I1,基于backward warping的方法需要从我们希望进行合成的帧It的角度近似中间流Ft-> 0,Ft-> 1。双向流的方法过于繁杂,无法达到实时。此外,也不能很好地消除运动边界伪影问题。

为了解决这些问题,本文首先提出了一个名为IFNet的专门高效的中间流网络,以直接估算中间流。 IFNet采取从粗到精的策略,并逐步提高分辨率。 它通过以递增分辨率运行的连续IFBlock迭代地更新流场。

如上图所示,RIFE是第一个基于流的实时VFI算法,以每秒30帧的速度处理720p视频。 实验表明,RIFE可以在多个公共基准上取得令人印象深刻的性能。

RIFE:

A.Pipeline Overview
在这里插入图片描述
如图2所示,给定两帧输入 I 0 I_{0} I0 I 1 I_{1} I1,VFI的目的是生成 t − > ( 0 , 1 ) t->(0,1) t>(0,1)的中间帧 I t ^ \hat{I_{t}} It^。首先,本文算法通过将输入帧输入到IFNet来直接估计中间流 F t − > 0 F_{t->0} Ft>0,然后使用线性运动假设近似 F t − > 1 F_{t->1} Ft>1
在这里插入图片描述
其次,通过backward warping输入帧得到两个粗略结果 I ^ 0 − > t \hat{I}_{0->t} I^0>t I ^ 1 − > t \hat{I}_{1->t} I^1>t。 为了消除warp后的伪像,本文使用FusionNet等编解码器将输入帧,近似流和warp后帧馈入融合过程,以生成最终的插帧图像。

B.Efficient Architecture Design

RIFE有两个主要组成部分:

(1)使用IFNet进行有效的中间流估算。

(2)使用FusionNet对warp后的帧进行融合处理。

(1)IFNet:
在这里插入图片描述
网络结构上图所示,IFNet采用了“Coarse-to-Fine”方式逐渐提升分辨率,也就是先在低分辨率上计算粗糙的光流,然后在高分辨率上计算精细的光流。主要组成部分就是IFBlock,降采样和上采样模块。

在这里插入图片描述
在上图中,本文提供了IFNet的视觉结果,并将它们与由预训练的LiteFlowNet 成的线性组合双向光流进行比较。可以看出,IFNet产生清晰而清晰的运动边界,而线性组合流会遇到像素重叠以及运动边界周围模糊的问题。

(2)Fusion process:

通过IFNet和对输入帧backward warping得到两个粗略结果 I ^ 0 − > t \hat{I}_{0->t} I^0>t I ^ 1 − > t \hat{I}_{1->t} I^1>t后,为降低仿射结果的伪影问题,作者提出了如下的调整与融合方式:
在这里插入图片描述
其中 M M M是用于融合两个warp帧的软融合图, Δ \Delta Δ是用于细化图像细节的重建残差项。

融合网络包括上下文提取器ContextNet和具有类似于U-Net的编解-码器架构的FusionNet。 FusionNet的上下文提取器和编码器部分具有相似的架构,由4个步幅为2的ResNet块组成。FusionNet的解码器部分具有四个转置卷积层。 使用sigmoid形函数来限制FusionNet的输出。

结合代码,具体来说:

class ContextNet(nn.Module):def __init__(self):super(ContextNet, self).__init__()self.conv1 = ResBlock(3, c)self.conv2 = ResBlock(c, 2*c)self.conv3 = ResBlock(2*c, 4*c)self.conv4 = ResBlock(4*c, 8*c)def forward(self, x, flow):x = self.conv1(x)f1 = warp(x, flow)x = self.conv2(x)flow = F.interpolate(flow, scale_factor=0.5, mode="bilinear",align_corners=False) * 0.5f2 = warp(x, flow)x = self.conv3(x)flow = F.interpolate(flow, scale_factor=0.5, mode="bilinear",align_corners=False) * 0.5f3 = warp(x, flow)x = self.conv4(x)flow = F.interpolate(flow, scale_factor=0.5, mode="bilinear",align_corners=False) * 0.5f4 = warp(x, flow)return [f1, f2, f3, f4]

ContextNet主要做的:

1.对原始输入图像 I 0 I_{0} I0 I 1 I_{1} I1通过上下文提取器分别提取金字塔上下文特征,文中表示为 C 0 C_{0} C0 C 1 C_{1} C1

2.对以上特征和对应的IFNet估计流一起使用backward warping,得到估计的中间流以生成对齐的金字塔特征 C 0 − > t C_{0->t} C0>t C 1 − > t C_{1->t} C1>t;每个金字塔上下文特征包含4个尺度的特征 f 1 f_{1} f1-> f 4 f_{4} f4

class FusionNet(nn.Module):def __init__(self):super(FusionNet, self).__init__()self.down0 = ResBlock(8, 2*c)self.down1 = ResBlock(4*c, 4*c)self.down2 = ResBlock(8*c, 8*c)self.down3 = ResBlock(16*c, 16*c)self.up0 = deconv(32*c, 8*c)self.up1 = deconv(16*c, 4*c)self.up2 = deconv(8*c, 2*c)self.up3 = deconv(4*c, c)self.conv = nn.Conv2d(c, 4, 3, 1, 1)def forward(self, img0, img1, flow, c0, c1, flow_gt):warped_img0 = warp(img0, flow)warped_img1 = warp(img1, -flow)if flow_gt == None:warped_img0_gt, warped_img1_gt = None, Noneelse:warped_img0_gt = warp(img0, flow_gt[:, :2])warped_img1_gt = warp(img1, flow_gt[:, 2:4])s0 = self.down0(torch.cat((warped_img0, warped_img1, flow), 1))s1 = self.down1(torch.cat((s0, c0[0], c1[0]), 1))s2 = self.down2(torch.cat((s1, c0[1], c1[1]), 1))s3 = self.down3(torch.cat((s2, c0[2], c1[2]), 1))x = self.up0(torch.cat((s3, c0[3], c1[3]), 1))x = self.up1(torch.cat((x, s2), 1))x = self.up2(torch.cat((x, s1), 1))x = self.up3(torch.cat((x, s0), 1))x = self.conv(x)return x, warped_img0, warped_img1, warped_img0_gt, warped_img1_gt

然后,将另外两个输入:warp后的的帧和IFNet估计的中间流输入到FusionNet,将FusionNet编码器部分中每个块的输出与相应的对齐金字塔特征连接起来,然后馈入下一个块。最后输出的x包含4个通,其中前3个通道是生成的重建残差 Δ \Delta Δ,第4个是融合图M。其他返回的4个warped_img0, warped_img1, warped_img0_gt, warped_img1_gt主要是被下面所介绍的Leakage Distillation使用。

C.Leakage Distillation for IFNet

为了使得IFNet预测的中间流结果比较准,采用一个预训练的光流模型提供额外的中间流信息作为监督信息训练。
在这里插入图片描述

Experiments:

1.UCF101, Vimeo90K, Middlebury OTHER set and HD benchmarks:
在这里插入图片描述
2.Increase model complexity:
在这里插入图片描述
3.Ablation Study:
在这里插入图片描述

这篇关于RIFE-实时视频插帧算法 | RIFE:Real-Time Intermediate Flow Estimation for Video Frame Interpolation的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!



http://www.chinasem.cn/article/379632

相关文章

openCV中KNN算法的实现

《openCV中KNN算法的实现》KNN算法是一种简单且常用的分类算法,本文主要介绍了openCV中KNN算法的实现,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的... 目录KNN算法流程使用OpenCV实现KNNOpenCV 是一个开源的跨平台计算机视觉库,它提供了各

基于Python和MoviePy实现照片管理和视频合成工具

《基于Python和MoviePy实现照片管理和视频合成工具》在这篇博客中,我们将详细剖析一个基于Python的图形界面应用程序,该程序使用wxPython构建用户界面,并结合MoviePy、Pill... 目录引言项目概述代码结构分析1. 导入和依赖2. 主类:PhotoManager初始化方法:__in

springboot+dubbo实现时间轮算法

《springboot+dubbo实现时间轮算法》时间轮是一种高效利用线程资源进行批量化调度的算法,本文主要介绍了springboot+dubbo实现时间轮算法,文中通过示例代码介绍的非常详细,对大家... 目录前言一、参数说明二、具体实现1、HashedwheelTimer2、createWheel3、n

用js控制视频播放进度基本示例代码

《用js控制视频播放进度基本示例代码》写前端的时候,很多的时候是需要支持要网页视频播放的功能,下面这篇文章主要给大家介绍了关于用js控制视频播放进度的相关资料,文中通过代码介绍的非常详细,需要的朋友可... 目录前言html部分:JavaScript部分:注意:总结前言在javascript中控制视频播放

Python基于wxPython和FFmpeg开发一个视频标签工具

《Python基于wxPython和FFmpeg开发一个视频标签工具》在当今数字媒体时代,视频内容的管理和标记变得越来越重要,无论是研究人员需要对实验视频进行时间点标记,还是个人用户希望对家庭视频进行... 目录引言1. 应用概述2. 技术栈分析2.1 核心库和模块2.2 wxpython作为GUI选择的优

SpringBoot实现MD5加盐算法的示例代码

《SpringBoot实现MD5加盐算法的示例代码》加盐算法是一种用于增强密码安全性的技术,本文主要介绍了SpringBoot实现MD5加盐算法的示例代码,文中通过示例代码介绍的非常详细,对大家的学习... 目录一、什么是加盐算法二、如何实现加盐算法2.1 加盐算法代码实现2.2 注册页面中进行密码加盐2.

Java时间轮调度算法的代码实现

《Java时间轮调度算法的代码实现》时间轮是一种高效的定时调度算法,主要用于管理延时任务或周期性任务,它通过一个环形数组(时间轮)和指针来实现,将大量定时任务分摊到固定的时间槽中,极大地降低了时间复杂... 目录1、简述2、时间轮的原理3. 时间轮的实现步骤3.1 定义时间槽3.2 定义时间轮3.3 使用时

python中time模块的常用方法及应用详解

《python中time模块的常用方法及应用详解》在Python开发中,时间处理是绕不开的刚需场景,从性能计时到定时任务,从日志记录到数据同步,时间模块始终是开发者最得力的工具之一,本文将通过真实案例... 目录一、时间基石:time.time()典型场景:程序性能分析进阶技巧:结合上下文管理器实现自动计时

如何通过Golang的container/list实现LRU缓存算法

《如何通过Golang的container/list实现LRU缓存算法》文章介绍了Go语言中container/list包实现的双向链表,并探讨了如何使用链表实现LRU缓存,LRU缓存通过维护一个双向... 目录力扣:146. LRU 缓存主要结构 List 和 Element常用方法1. 初始化链表2.

golang字符串匹配算法解读

《golang字符串匹配算法解读》文章介绍了字符串匹配算法的原理,特别是Knuth-Morris-Pratt(KMP)算法,该算法通过构建模式串的前缀表来减少匹配时的不必要的字符比较,从而提高效率,在... 目录简介KMP实现代码总结简介字符串匹配算法主要用于在一个较长的文本串中查找一个较短的字符串(称为