From a27fd674ece5b066fcaf8c5f75dc6471899d5609 Mon Sep 17 00:00:00 2001 From: hzwer <598460606@163.com> Date: Fri, 13 Nov 2020 11:00:33 +0800 Subject: [PATCH] Update README.md --- README.md | 9 ++------- 1 file changed, 2 insertions(+), 7 deletions(-) diff --git a/README.md b/README.md index f9095d1..350ab21 100644 --- a/README.md +++ b/README.md @@ -1,6 +1,7 @@ # RIFE ## [arXiv](https://arxiv.org/abs/2011.06294) -**We are working on arranging our training code and other scripts.** +**We are working on arranging our training code and other scripts.** +Our method support CPU and GPU with CUDA. ## Abstract We propose a real-time intermediate flow estimation algorithm (RIFE) for video frame interpolation (VFI). Most existing methods first estimate the bi-directional optical flows, and then linearly combine them to approximate intermediate flows, leading to artifacts around motion boundaries. We design an intermediate flow model named IFNet that can directly estimate the intermediate flows from coarse to fine. We then warp the input frames according to the estimated intermediate flows and employ a fusion process to compute final results. Based on our proposed leakage distillation, RIFE can be trained end-to-end and achieve excellent performance. Experiments demonstrate that RIFE is significantly faster than existing flow-based VFI methods and achieves state-of-the-art index on several benchmarks. @@ -22,7 +23,6 @@ pip3 install opencv-python ``` $ python3 inference.py --img /path/to/image_0 /path/to/image_1 // You will get an out.png. -// You can use cpu to have a try. ``` If you find this repository useful for your research, please cite the following paper: @@ -37,13 +37,8 @@ If you find this repository useful for your research, please cite the following ## Reference [ARFlow](https://github.com/lliuz/ARFlow) - [pytorch-liteflownet](https://github.com/sniklaus/pytorch-liteflownet) - [RAFT](https://github.com/princeton-vl/RAFT) - [DAIN](https://github.com/baowenbo/DAIN) - [CAIN](https://github.com/myungsub/CAIN) - [AdaCoF-pytorch](https://github.com/HyeongminLEE/AdaCoF-pytorch)