HolyWu / vs-rife

RIFE function for VapourSynth
MIT License
102 stars 9 forks source link
pytorch rife tensorrt vapoursynth

RIFE

Real-Time Intermediate Flow Estimation for Video Frame Interpolation, based on https://github.com/hzwer/Practical-RIFE.

Dependencies

trt requires additional packages:

To install the latest nightly build of PyTorch and Torch-TensorRT, run:

pip install -U packaging setuptools wheel
pip install --pre -U torch torchvision --index-url https://download.pytorch.org/whl/nightly/cu124
pip install --no-deps --pre -U torch_tensorrt --index-url https://download.pytorch.org/whl/nightly/cu124
pip install -U tensorrt-cu12 tensorrt-cu12_bindings tensorrt-cu12_libs --extra-index-url https://pypi.nvidia.com

Installation

pip install -U vsrife
python -m vsrife

Usage

from vsrife import rife

ret = rife(clip)

See __init__.py for the description of the parameters.