Ma/video2x
1
0
Fork 0
mirror of https://github.com/k4yt3x/video2x.git synced 2026-04-14 14:47:32 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat(rife): add support for rife v4.25 & v4.26 (#1304)

Browse Source 
* chore(deps): update rife submodule
* update: add support for rife v4.25 & v4.26
* style(*): format code with clang-format

---------

Signed-off-by: k4yt3x <i@k4yt3x.com>
Co-authored-by: k4yt3x <i@k4yt3x.com>
This commit is contained in:
Dawei Feng
2025-01-22 01:37:35 +08:00
committed by GitHub
parent 947788225e
commit 9d4b21b4c1
11 changed files with 1193 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
src/interpolator_rife.cpp
View File

@@ -50,20 +50,38 @@ int InterpolatorRIFE::init(AVCodecContext* dec_ctx, AVCodecContext* enc_ctx, AVB
// Automatically infer the RIFE model generation based on the model name
bool rife_v2 = false;
bool rife_v4 = false;
int rife_padding = 32;
if (model_name_.find(STR("rife-v2")) != fsutils::StringType::npos) {
rife_v2 = true;
} else if (model_name_.find(STR("rife-v3")) != fsutils::StringType::npos) {
rife_v2 = true;
} else if (model_name_.find(STR("rife-v4")) != fsutils::StringType::npos) {
rife_v4 = true;
if (model_name_.find(STR("rife-v4.25")) != fsutils::StringType::npos) {
rife_padding = 64;
}
if (model_name_.find(STR("rife-v4.25-lite")) != fsutils::StringType::npos) {
rife_padding = 128;
}
if (model_name_.find(STR("rife-v4.26")) != fsutils::StringType::npos) {
rife_padding = 64;
}
} else if (model_name_.find(STR("rife")) == fsutils::StringType::npos) {
logger()->critical("Failed to infer RIFE model generation from model name");
return -1;
}
// Create a new RIFE instance
rife_ =
new RIFE(gpuid_, tta_mode_, tta_temporal_mode_, uhd_mode_, num_threads_, rife_v2, rife_v4);
rife_ = new RIFE(
gpuid_,
tta_mode_,
tta_temporal_mode_,
uhd_mode_,
num_threads_,
rife_v2,
rife_v4,
rife_padding
);
// Store the time bases
in_time_base_ = dec_ctx->time_base;