OpenSourceJoker/video2x
3
0
Fork 0
mirror of https://github.com/k4yt3x/video2x.git synced 2024-12-28 06:59:11 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix(realesrgan): fixed RealESRGAN model name variable

Browse Source 
Signed-off-by: k4yt3x <i@k4yt3x.com>
This commit is contained in:
k4yt3x 2024-11-02 00:00:00 +00:00
parent 406a97f360
commit bb74144070
No known key found for this signature in database
2 changed files with 11 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
include/libvideo2x/libvideo2x.h
View File

@ -56,7 +56,7 @@ struct RealESRGANConfig {
bool tta_mode;
int scaling_factor;
#ifdef _WIN32
const wchar_t *model_path;
const wchar_t *model_name;
#else
const char *model_name;
#endif

16
src/video2x.cpp
View File

@ -80,8 +80,12 @@ struct Arguments {
int out_height = 0;
// RealESRGAN options
#ifdef _WIN32
std::wstring model_name;
#else
std::string model_name;
#endif
int gpuid = 0;
std::filesystem::path model_path;
int scaling_factor = 0;
};
@ -362,9 +366,9 @@ int main(int argc, char **argv) {
if (vm.count("model")) {
#ifdef _WIN32
arguments.model_path = std::filesystem::path(vm["model"].as<std::wstring>());
arguments.model_name = std::filesystem::path(vm["model"].as<std::wstring>());
#else
arguments.model_path = vm["model"].as<std::string>();
arguments.model_name = vm["model"].as<std::string>();
#endif
if (!is_valid_realesrgan_model(vm["model"].as<std::string>())) {
spdlog::error(
@ -401,7 +405,7 @@ int main(int argc, char **argv) {
#else
} else if (arguments.filter_type == "realesrgan") {
#endif
if (arguments.scaling_factor == 0 || arguments.model_path.empty()) {
if (arguments.scaling_factor == 0 || arguments.model_name.empty()) {
spdlog::error("Error: For realesrgan, scaling factor (-r) and model (-m) are required."
);
return 1;
@ -500,9 +504,9 @@ int main(int argc, char **argv) {
filter_config.config.realesrgan.tta_mode = false;
filter_config.config.realesrgan.scaling_factor = arguments.scaling_factor;
#ifdef _WIN32
filter_config.config.realesrgan.model_name = arguments.model_path.c_str();
filter_config.config.realesrgan.model_name = arguments.model_name.c_str();
#else
filter_config.config.realesrgan.model_name = arguments.model_path.c_str();
filter_config.config.realesrgan.model_name = arguments.model_name.c_str();
filter_config.config.realesrgan.model_name = "realesr-animevideov4";
#endif
}