splinter21 / rife-ncnn-vulkan-python

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

RIFE ncnn Vulkan Python

CI

Introduction

rife-ncnn-vulkan is nihui's ncnn implementation of Real-World Super-Resolution via Kernel Estimation and Noise Injection super resolution.

rife-ncnn-vulkan-python wraps rife-ncnn-vulkan project by SWIG to make it easier to integrate rife-ncnn-vulkan with existing python projects.

Downloads

Linux/Windos/Mac X86_64 build releases are available now. However, for Linux you may need to compile it yourself if you are using distro with GLIBC < 2.29 .

Build

First, you have to install python, python development package (Python native development libs in Visual Studio), vulkan SDK and SWIG on your platform. And then:

Linux

git clone https://github.com/ArchieMeng/rife-ncnn-vulkan-python.git
cd rife-ncnn-vulkan-python
git submodule update --init --recursive
cmake -B build src
cd build
make

Windows

I used Visual Studio 2019 and msvc v142 to build this project for Windows.

Install visual studio and open the project directory, and build. Job done.

The only problem on Windows is that, you cannot use CMake for Windows to generate the Visual Studio solution file and build it. This will make the lib crash on loading.

The only way is use Visual Studio to open the project as directory, and build it from Visual Studio.

About RIFE

RIFE (Real-Time Intermediate Flow Estimation for Video Frame Interpolation)

https://github.com/hzwer/arXiv2020-RIFE

Huang, Zhewei and Zhang, Tianyuan and Heng, Wen and Shi, Boxin and Zhou, Shuchang

https://rife-vfi.github.io

https://arxiv.org/abs/2011.06294

Usages

Example Program

from PIL import Image
from rife_ncnn_vulkan import RIFE

im0, im1 = Image.open("0.png"), Image.open("1.png")
rife = RIFE(0)
im = rife.process(im0, im1)
im.save("interframe.png")

If you encounter a crash or error, try upgrading your GPU driver:

Model

model upstream version
rife 1.2
rife-HD 1.5
rife-UHD 1.6
rife-anime 1.8
rife-v2 2.0
rife-v2.3 2.3
rife-v2.4 2.4

Sample Images

Original Image

origin0 origin1

Interpolate with rife rife-anime model

rife

Interpolate with rife rife-anime model + TTA-s

rife

Original RIFE Project

Other Open-Source Code Used

About

License:MIT License


Languages

Language:C 78.1%Language:C++ 18.2%Language:CMake 3.0%Language:Python 0.5%Language:SWIG 0.1%