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

feat(realcugan): add support for Real-CUGAN ncnn Vulkan (#1268)

Browse Source 
Signed-off-by: k4yt3x <i@k4yt3x.com>
This commit is contained in:
K4YT3X
2024-12-20 21:58:19 -05:00
committed by GitHub
parent e1e8d64056
commit 127d9e0019
54 changed files with 1469 additions and 30 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
include/libvideo2x/processor.h
View File

@@ -23,6 +23,7 @@ enum class ProcessorType {
None,
Libplacebo,
RealESRGAN,
RealCUGAN,
RIFE,
};
@@ -35,6 +36,13 @@ struct RealESRGANConfig {
fsutils::StringType model_name;
};
struct RealCUGANConfig {
bool tta_mode = false;
int num_threads = 1;
int syncgap = 3;
fsutils::StringType model_name;
};
struct RIFEConfig {
bool tta_mode = false;
bool tta_temporal_mode = false;
@@ -49,9 +57,10 @@ struct ProcessorConfig {
int width = 0;
int height = 0;
int scaling_factor = 0;
int noise_level = -1;
int frm_rate_mul = 0;
float scn_det_thresh = 0.0f;
std::variant<LibplaceboConfig, RealESRGANConfig, RIFEConfig> config;
std::variant<LibplaceboConfig, RealESRGANConfig, RealCUGANConfig, RIFEConfig> config;
};
class Processor {