用AI脑补慢动作，30帧秒变120帧，轻松看清“松果糖豆五连鞭”动作细节-技术圈


来源：HyberAI超神经

本文约2400字，建议阅读8分钟
本文带你了解Super SloMo教程。

如今，人们对于视频帧率的追求越来越高，因为高帧率视频更加顺滑、流畅，能极大地提升人们的观看体验。

现有相机拍摄的视频帧率，也从 24 FPS（Frames Per Second) 不断提升到 60 FPS，再到 240 FPS 甚至更高。

在绝大多数电影帧率 24 的当下，李安的《双子杀手》以 120 的帧率实现电影技术的革新

但是，高帧率的摄像设备，对于内存需求非常大，而且成本高昂，尚不能普及。为了在没有专业设备的情况下，能够得到高帧率的视频，视频插帧技术应运而生。

而英伟达提出的 AI「脑补」大法 Super SloMo，则在众多视频插帧技术中一骑绝尘，即使帧率仅仅为 30 帧的视频，也可以 Super SloMo 补成 60 帧、240 帧甚至更高。

传统插帧方法优劣势一览

为了更好地理解 Super SloMo，我们首先来了解一下现有的较为传统的视频插帧技术。

帧采样

帧采样就是用关键帧来做补偿帧，其实质就是拉长每一个关键帧的显示时间，相当于并没有插帧。除了获得文件属性更高的帧率和在同视频质量下更大的文件体积外，不会带来任何视觉观感上的提升。

优点：帧采样消耗资源少，速度快。

缺点：可能会使得视频看起来不是很流畅。

帧混合

帧混合，顾名思义，就是提升前后关键帧的透明度，再将其混合成一个新的帧，来填补空缺。

优点：计算所需时长短。

缺点：效果不佳。由于只是简单的把原关键帧变成半透明状，运动物体轮廓在前后两帧交叠的时候，就会产生明显的模糊场景，对视频的视觉效果流畅程度提升很小。

运动补偿

运动补偿（Motion Estimation and Motion Compensation，简称 MEMC），其原理是在水平和垂直两个方向上，对两帧之间的差异寻找出现运动的块，通过分析图像块的运动趋势，再辅以计算得到中间帧。

MEMC 主要被应用于电视、显示器和移动端，提升视频帧率，给观众更加流畅的观感。

优点：减小运动抖动，减弱画面拖尾与虚影，提升画面清晰度。

缺点：对于运动物体背景比较复杂的情况，就会出现物体边缘运动的 bug。

光流法

光流法是计算机视觉研究中的一个重要方向，其根据上下帧来推断像素移动的轨迹，自动生成新的空缺帧。有点类似于运动模糊计算方法。

优点：画面更流畅、卡顿感弱。

缺点：计算量大, 耗时长；对光线敏感，在光线变化较大的情况下，容易出现画面紊乱的错误。

Super SloMo：AI 插帧法，堪称业界经典

在 2018 年计算机顶会 CVPR 上，英伟达发布的论文《Super SloMo: High Quality Estimation of Multiple Intermediate Frames for Video Interpolation》中，提出了 Super SloMo，在业界引起广泛关注。

《超级慢动作：用于视频多个中间帧插值的的高质量估计》论文地址：

https://arxiv.org/pdf/1712.00080.pdf

Super SloMo 不同于传统方法，它利用深度神经网络来实现补帧，基本思路为：使用大量普通视频与慢动作视频进行训练，然后让神经网络学会推理，根据正常视频生成高质量的超级慢动作视频。

Super SloMo 方法框架图包括光流计算模块（左）与特定时间流插值模块（右）

Super SloMo 团队提出的方法，其整个框架依赖于两个全卷积神经网络 U-Net。

首先，用一个 U-Net 来计算相邻输入图像之间的双向光流。然后，在每个时间步长上对这些光流进行线性拟合，以近似中间帧的双向光流。

为了解决运动边界出现伪影的问题，使用另一个 U-Net 来对近似的光流进行改善，并且预测柔性可见性映射关系。最后，将输入的两张图像进行扭曲和线性融合，从而形成中间帧。

此外，Super SloMo 的光流计算网络和插值网络的参数，都不依赖于被插值帧的特定时间步长（时间步长被做为网络的输入）。因此，它可以并行地在两帧之间的任意时间步长上插帧，从而突破了很多单帧插值方法的局限性。

原始 SloMo 视频（上）Super SloMo 补帧后的超级慢动作视频（下）

作者表示，使用他们未经优化的 PyTorch 代码，在单个 NVIDIA GTX 1080Ti 和 Tesla V100 GPU 上，生成 7 个分辨率为 1280*720 的中间帧，分别只需要 0.97 秒和 0.79 秒。

为了训练网络，作者从 YouTube 和手持摄像机上收集了多个 240 帧的视频。总计收集了 1100 个视频段，由 30 万张 1080×720 分辨率的独立的视频帧组成。这些视频从室内到室外，从静态相机到动态相机，从日常活动到专业运动，包含了各类场景。

之后在其它数据集上对模型进行了验证，结果表明，该研究在这些数据集上比现有的方法性能上有了显著的提高。

跟着教程，一键实现 Super SloMo

虽然英伟达这一论文的作者尚未公开数据集和代码，不过，高手在民间，GitHub 上一位名为为 avinashpaliwal 的用户，已经开源了自己对 Super SloMo 的 PyTorch 实现，其结果与论文描述相差无几。

该项目具体信息如下：

Super SloMo 超级慢动作镜头补帧

运行环境：PyTorch 0.4.1

语言版本：Python 3.6

训练可视化：TensorboardX

训练数据集：Adobe 240 fps

项目地址：

https://github.com/avinashpaliwal/Super-SloMo

由于模型训练和测试是在 PyTorch 0.4.1 和 CUDA 9.2 上完成的，所以安装这两个软件必不可少，另外你还需要有一张 NVIDIA 的显卡。

此外，模型不能直接使用视频训练，因此还需要安装 ffmpeg 从视频中提取帧。这些准备工作一切就绪后，方可下载 adobe 240fps 数据集进行训练。

不过，你也可以不需要准备这些，做一个安安静静的「伸手党」，一键实现 Super SloMo。

我们在国内的机器学习算力容器服务平台（https://openbayes.com）找到了相应的教程。

从数据集到代码再到算力，一应俱全，即使是小白，也可以轻松上手。

教程链接：

https://openbayes.com/console/openbayes/containers/xQIPlDQ0GyD/overview

教程使用指南

首先注册并登录 https://openbayes.com/，在「公共资源」菜单下的「公共教程」中，选择本教程——「Super-SloMo超级慢动作镜头的PyTorch实现」。

教程中样例展示文件是 Super-SloMo.ipynb，运行这个文件，会安装环境，并展示最终补帧而来的超级慢动作效果。

大家也可以使用自己的视频素材，将下方生成代码中的 lightning-dick-clip.mp4 改为你的视频文件名。

属性「scale」用于控制生成视频的倍速，比如：设为 4，即 4 倍速慢放。

生成代码：

!python3 'Super-SloMo/eval.py' \    'lightning-dick-clip.mp4' \    --checkpoint='/openbayes/input/input0/SuperSloMo.ckpt' \    --output='output-tmp.mp4' \    --scale=4print('Done')

转换视频格式代码：

!ffmpeg -i output-tmp.mp4 -vcodec libx264 -acodec aac output.mp4

该教程中，使用了网络上的某段视频，进行 Super SloMo 插帧，得到如下结果：

4倍速慢放，看清武术动作的每一步

目前，该平台还能薅羊毛，每周都赠送 vGPU 的使用时长，人人都可轻松完成，赶快动手试试吧！

参考资料：

https://arxiv.org/pdf/1712.00080.pdf

http://jianghz.me/projects/superslomo/https://zhuanlan.zhihu.com/p/86426432

编辑：王菁

校对：林亦霖