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

* 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>

CMakeLists.txt

@@ -212,7 +212,7 @@ else()
         option(WITH_LAYER_input "" ON)
         option(WITH_LAYER_log "" OFF)
         option(WITH_LAYER_lrn "" OFF)
-        option(WITH_LAYER_memorydata "" OFF)
+        option(WITH_LAYER_memorydata "" ON)
         option(WITH_LAYER_mvn "" OFF)
         option(WITH_LAYER_pooling "" ON)
         option(WITH_LAYER_power "" OFF)
@@ -287,6 +287,7 @@ else()
         option(WITH_LAYER_deconvolutiondepthwise3d "" OFF)
         option(WITH_LAYER_einsum "" OFF)
         option(WITH_LAYER_deformableconv2d "" OFF)
+        option(WITH_LAYER_gridsample "" ON)
 
         add_subdirectory(third_party/ncnn)
     endif()

src/interpolator_rife.cpp

@@ -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;

tools/video2x/src/argparse.cpp

@@ -180,7 +180,8 @@ int parse_args(
             ("rife-model", PO_STR_VALUE<video2x::fsutils::StringType>()
                 ->default_value(STR("rife-v4.6"), "rife-v4.6")->notifier(validate_rife_model_name),
                 "Name of the RIFE model to use (rife, rife-HD, rife-UHD, rife-anime, rife-v2, "
-                "rife-v2.3, rife-v2.4, rife-v3.0, rife-v3.1, rife-v4, rife-v4.6)")
+                "rife-v2.3, rife-v2.4, rife-v3.0, rife-v3.1, rife-v4, rife-v4.6, rife-v4.25, "
+                "rife-v4.25-lite, rife-v4.26)")
             ("rife-uhd", "Enable Ultra HD mode")
         ;
         // clang-format on

tools/video2x/src/validators.cpp

@@ -62,13 +62,17 @@ void validate_rife_model_name(const video2x::fsutils::StringType& model_name) {
         STR("rife-v3.1"),
         STR("rife-v4"),
         STR("rife-v4.6"),
+        STR("rife-v4.25"),
+        STR("rife-v4.25-lite"),
+        STR("rife-v4.26"),
     };
     if (valid_rife_models.count(model_name) == 0) {
         throw po::validation_error(
             po::validation_error::invalid_option_value,
             "rife-model",
             "RIFE model must be one of: rife, rife-HD, rife-UHD, rife-anime, rife-v2, rife-v2.3, "
-            "rife-v2.4, rife-v3.0, rife-v3.1, rife-v4, rife-v4.6"
+            "rife-v2.4, rife-v3.0, rife-v3.1, rife-v4, rife-v4.6, rife-v4.25, rife-v4.25-lite, "
+            "rife-v4.26"
         );
     }
 }