Super-SloMo 视频多帧插帧项目

这个 notebook 是基于 Super-SloMo 超级慢动作项目而建立。基于 PyTorch 实现的「Super SloMo: High Quality Estimation of Multiple Intermediate Frames for Video Interpolation by Jiang H., Sun D., Jampani V., Yang M., Learned-Miller E. and Kautz J」。

本项目只提供了如何生成超级慢动作视频，不包含训练等相关操作。生成基于 Adobe240fps 预训练模型。

依次执行下列代码：

准备环境

import os
mox.file.copy_parallel('obs://obs-aigallery-zc/clf/code/Super-SloMo.zip','Super-SloMo.zip')
os.system('unzip Super-SloMo.zip')
os.chdir('Super-SloMo')

!pip install click
!pip install aubio ffmpeg

如何使用：

通过拖拽把你要变慢的视频放入左侧的文件列表中。或者通过上方的上传按钮也可以。
然后修改下方的 lightning-dick-clip.mp4 为你的视频文件名，需要带上 mp4 后缀，不要去掉最外层的引号。

属性 scale 用于控制生成视频的倍速，例如 4 就是 4 倍速放慢

开始生成

!python3 'Super-SloMo/eval.py' \
    'lightning-dick-clip.mp4' \
    --checkpoint='model/SuperSloMo.ckpt' \
    --output='output-tmp.mp4' \
    --scale=4

print('Done')

# 转换视频格式
!ffmpeg -i output-tmp.mp4 -vcodec libx264 -acodec aac output.mp4

!rm -f output-tmp.mp4
from IPython.display import HTML
from base64 import b64encode
mp4 = open('output.mp4','rb').read()
data_url = "data:video/mp4;base64," + b64encode(mp4).decode()
HTML("""
<video width=600 controls>
      <source src="%s" type="video/mp4">
</video>
""" % data_url)

结果

上方的预览就就是最终的输出结果，你也可以直接在左侧文件列表中下载对应的输出文件