Skip to content

Commit

Permalink
add rife-v4.17 (#40)
Browse files Browse the repository at this point in the history
  • Loading branch information
TNTwise authored May 25, 2024
1 parent 145d400 commit 42a3ba5
Show file tree
Hide file tree
Showing 7 changed files with 963 additions and 9 deletions.
2 changes: 1 addition & 1 deletion .github/workflows/CI.yml
Original file line number Diff line number Diff line change
Expand Up @@ -68,7 +68,7 @@ jobs:
ninja -C build
macOS:
runs-on: macos-latest
runs-on: macos-12

env:
CC: /usr/local/opt/llvm/bin/clang
Expand Down
9 changes: 5 additions & 4 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -73,12 +73,13 @@ By default models are exported with ensemble=False and Fast=True
- 50 = rife-v4.15-lite (ensemble=True)
- 51 = rife-v4.16-lite (ensemble=False)
- 52 = rife-v4.16-lite (ensemble=True)

- 53 = rife-v4.17 (ensemble=False)
- 54 = rife-v4.17 (ensemble=True)
## My experimental custom models (only works with 2x)

- 53 = sudo_rife4 (ensemble=False / fast=True)
- 54 = sudo_rife4 (ensemble=True / fast=False)
- 55 = sudo_rife4 (ensemble=True / fast=True)
- 55 = sudo_rife4 (ensemble=False / fast=True)
- 56 = sudo_rife4 (ensemble=True / fast=False)
- 57 = sudo_rife4 (ensemble=True / fast=True)

- factor_num, factor_den: Factor of target frame rate. For example `factor_num=5, factor_den=2` will multiply input clip FPS by 2.5. Only rife-v4 model supports custom frame rate.

Expand Down
14 changes: 10 additions & 4 deletions RIFE/plugin.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -203,8 +203,8 @@ static void VS_CC rifeCreate(const VSMap* in, VSMap* out, [[maybe_unused]] void*
if (err)
d->skipThreshold = 60.0;

if (model < 0 || model > 55)
throw "model must be between 0 and 55 (inclusive)";
if (model < 0 || model > 57)
throw "model must be between 0 and 57 (inclusive)";

if (factorNum < 1)
throw "factor_num must be at least 1";
Expand Down Expand Up @@ -445,12 +445,18 @@ static void VS_CC rifeCreate(const VSMap* in, VSMap* out, [[maybe_unused]] void*
modelPath += "/rife-v4.16_lite_ensembleTrue";
break;
case 53:
modelPath += "/sudo_rife4_ensembleFalse_fastTrue";
modelPath += "/rife-v4.17_ensembleFalse";
break;
case 54:
modelPath += "/sudo_rife4_ensembleTrue_fastFalse";
modelPath += "/rife-v4.17_ensembleTrue";
break;
case 55:
modelPath += "/sudo_rife4_ensembleFalse_fastTrue";
break;
case 56:
modelPath += "/sudo_rife4_ensembleTrue_fastFalse";
break;
case 57:
modelPath += "/sudo_rife4_ensembleTrue_fastTrue";
break;

Expand Down
Binary file added models/rife-v4.17_ensembleFalse/flownet.bin
Binary file not shown.
315 changes: 315 additions & 0 deletions models/rife-v4.17_ensembleFalse/flownet.param

Large diffs are not rendered by default.

Binary file added models/rife-v4.17_ensembleTrue/flownet.bin
Binary file not shown.
632 changes: 632 additions & 0 deletions models/rife-v4.17_ensembleTrue/flownet.param

Large diffs are not rendered by default.

0 comments on commit 42a3ba5

Please sign in to comment.