Skip to content

Commit

Permalink
Merge pull request #49 from TNTwise/master
Browse files Browse the repository at this point in the history
4.22-lite
  • Loading branch information
styler00dollar authored Aug 20, 2024
2 parents 35ac6d7 + c1c74b7 commit 08c941c
Show file tree
Hide file tree
Showing 4 changed files with 328 additions and 7 deletions.
7 changes: 4 additions & 3 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -85,12 +85,13 @@ By default models are exported with ensemble=False and Fast=True
- 62 = rife-v4.20 (ensemble=True)
- 63 = rife-v4.21 (ensemble=False)
- 64 = rife-v4.22 (ensemble=False)
- 65 = rife-v4.22-lite(ensemble=False)

## My experimental custom models (only works with 2x)

- 65 = sudo_rife4 (ensemble=False / fast=True)
- 66 = sudo_rife4 (ensemble=True / fast=False)
- 67 = sudo_rife4 (ensemble=True / fast=True)
- 66 = sudo_rife4 (ensemble=False / fast=True)
- 67 = sudo_rife4 (ensemble=True / fast=False)
- 68 = 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
11 changes: 7 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 > 67)
throw "model must be between 0 and 65 (inclusive)";
if (model < 0 || model > 68)
throw "model must be between 0 and 68 (inclusive)";

if (factorNum < 1)
throw "factor_num must be at least 1";
Expand Down Expand Up @@ -481,12 +481,15 @@ static void VS_CC rifeCreate(const VSMap* in, VSMap* out, [[maybe_unused]] void*
modelPath += "/rife-v4.22_ensembleFalse";
break;
case 65:
modelPath += "/sudo_rife4_ensembleFalse_fastTrue";
modelPath += "/rife-v4.22_lite_ensembleFalse";
break;
case 66:
modelPath += "/sudo_rife4_ensembleTrue_fastFalse";
modelPath += "/sudo_rife4_ensembleFalse_fastTrue";
break;
case 67:
modelPath += "/sudo_rife4_ensembleTrue_fastFalse";
break;
case 68:
modelPath += "/sudo_rife4_ensembleTrue_fastTrue";
break;

Expand Down
Binary file added models/rife-v4.22_lite_ensembleFalse/flownet.bin
Binary file not shown.
Loading

0 comments on commit 08c941c

Please sign in to comment.