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

chore(models): add the Real-ESRGAN general models (#1319)

Browse Source 
Signed-off-by: k4yt3x <i@k4yt3x.com>
This commit is contained in:
k4yt3x
2025-02-02 00:00:00 +00:00
parent 2c5a059d39
commit afa37f9e58
10 changed files with 195 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
src/filter_realesrgan.cpp
View File

@@ -17,12 +17,14 @@ FilterRealesrgan::FilterRealesrgan(
int gpuid,
bool tta_mode,
int scaling_factor,
int noise_level,
const fsutils::StringType model_name
)
: realesrgan_(nullptr),
gpuid_(gpuid),
tta_mode_(tta_mode),
scaling_factor_(scaling_factor),
noise_level_(noise_level),
model_name_(std::move(model_name)) {}
FilterRealesrgan::~FilterRealesrgan() {
@@ -35,10 +37,16 @@ int FilterRealesrgan::init(AVCodecContext* dec_ctx, AVCodecContext* enc_ctx, AVB
std::filesystem::path model_param_path;
std::filesystem::path model_bin_path;
fsutils::StringType param_file_name =
model_name_ + STR("-x") + fsutils::to_string_type(scaling_factor_) + STR(".param");
fsutils::StringType bin_file_name =
model_name_ + STR("-x") + fsutils::to_string_type(scaling_factor_) + STR(".bin");
fsutils::StringType param_file_name = model_name_;
fsutils::StringType bin_file_name = model_name_;
if (model_name_ == STR("realesr-generalv3") && noise_level_ > 0) {
param_file_name += STR("-wdn");
bin_file_name += STR("-wdn");
}
param_file_name += STR("-x") + fsutils::to_string_type(scaling_factor_) + STR(".param");
bin_file_name += STR("-x") + fsutils::to_string_type(scaling_factor_) + STR(".bin");
// Find the model paths by model name if provided
model_param_path = std::filesystem::path(STR("models")) / STR("realesrgan") / param_file_name;