From d279b1bc910cf334bfa34bb371745e319a23ebeb Mon Sep 17 00:00:00 2001 From: Sabih Rahamn Date: Sun, 25 Feb 2024 15:56:57 +0500 Subject: [PATCH] Changed default weight paths in yaml configurations. Added .yml file for main models. --- options/Download_paths.yml | 18 ++++++++++++++++++ options/finetune_realesrgan_x4plus.yml | 2 +- .../finetune_realesrgan_x4plus_pairdata.yml | 2 +- options/train_realesrgan_x2plus.yml | 2 +- options/train_realesrgan_x4plus.yml | 2 +- options/train_realesrnet_x2plus.yml | 2 +- 6 files changed, 23 insertions(+), 5 deletions(-) create mode 100644 options/Download_paths.yml diff --git a/options/Download_paths.yml b/options/Download_paths.yml new file mode 100644 index 000000000..68d4f6c1d --- /dev/null +++ b/options/Download_paths.yml @@ -0,0 +1,18 @@ +Model_Paths: + + RealESRGAN-x4plus: + Generator: https://github.com/xinntao/Real-ESRGAN/releases/download/v0.1.0/RealESRGAN_x4plus.pth + Discriminator: https://github.com/xinntao/Real-ESRGAN/releases/download/v0.2.2.3/RealESRGAN_x4plus_netD.pth + + RealESRGAN-x2plus: + Generator: https://github.com/xinntao/Real-ESRGAN/releases/download/v0.2.1/RealESRGAN_x2plus.pth + Discriminator: https://github.com/xinntao/Real-ESRGAN/releases/download/v0.2.2.3/RealESRGAN_x2plus_netD.pth + + RealESRNet-x4plus: + Generator: https://github.com/xinntao/Real-ESRGAN/releases/download/v0.1.1/RealESRNet_x4plus.pth + Discriminator: https://github.com/xinntao/Real-ESRGAN/releases/download/v0.2.2.3/RealESRGAN_x4plus_netD.pth + + RealESRNet-x2plus: + Generator: - + Discriminator: https://github.com/xinntao/Real-ESRGAN/releases/download/v0.2.2.3/RealESRGAN_x2plus_netD.pth + diff --git a/options/finetune_realesrgan_x4plus.yml b/options/finetune_realesrgan_x4plus.yml index aa9806570..1cfa8ed65 100644 --- a/options/finetune_realesrgan_x4plus.yml +++ b/options/finetune_realesrgan_x4plus.yml @@ -99,7 +99,7 @@ network_d: # path path: # use the pre-trained Real-ESRNet model - pretrain_network_g: experiments/pretrained_models/RealESRNet_x4plus.pth + pretrain_network_g: experiments/pretrained_models/RealESRGAN_x4plus.pth param_key_g: params_ema strict_load_g: true pretrain_network_d: experiments/pretrained_models/RealESRGAN_x4plus_netD.pth diff --git a/options/finetune_realesrgan_x4plus_pairdata.yml b/options/finetune_realesrgan_x4plus_pairdata.yml index db45d4d27..36e41504c 100644 --- a/options/finetune_realesrgan_x4plus_pairdata.yml +++ b/options/finetune_realesrgan_x4plus_pairdata.yml @@ -61,7 +61,7 @@ network_d: # path path: # use the pre-trained Real-ESRNet model - pretrain_network_g: experiments/pretrained_models/RealESRNet_x4plus.pth + pretrain_network_g: experiments/pretrained_models/RealESRGAN_x4plus.pth param_key_g: params_ema strict_load_g: true pretrain_network_d: experiments/pretrained_models/RealESRGAN_x4plus_netD.pth diff --git a/options/train_realesrgan_x2plus.yml b/options/train_realesrgan_x2plus.yml index 3c98a0f37..09b79b374 100644 --- a/options/train_realesrgan_x2plus.yml +++ b/options/train_realesrgan_x2plus.yml @@ -100,7 +100,7 @@ network_d: # path path: # use the pre-trained Real-ESRNet model - pretrain_network_g: experiments/pretrained_models/RealESRNet_x2plus.pth + pretrain_network_g: experiments/pretrained_models/RealESRGAN_x2plus.pth param_key_g: params_ema strict_load_g: true resume_state: ~ diff --git a/options/train_realesrgan_x4plus.yml b/options/train_realesrgan_x4plus.yml index 763199a35..51c4aa59c 100644 --- a/options/train_realesrgan_x4plus.yml +++ b/options/train_realesrgan_x4plus.yml @@ -99,7 +99,7 @@ network_d: # path path: # use the pre-trained Real-ESRNet model - pretrain_network_g: experiments/pretrained_models/RealESRNet_x4plus.pth + pretrain_network_g: experiments/pretrained_models/RealESRGAN_x4plus.pth param_key_g: params_ema strict_load_g: true resume_state: ~ diff --git a/options/train_realesrnet_x2plus.yml b/options/train_realesrnet_x2plus.yml index 81ee9ef16..6bd9070fc 100644 --- a/options/train_realesrnet_x2plus.yml +++ b/options/train_realesrnet_x2plus.yml @@ -90,7 +90,7 @@ network_g: # path path: - pretrain_network_g: experiments/pretrained_models/RealESRGAN_x4plus.pth + pretrain_network_g: experiments/pretrained_models/RealESRGAN_x2plus.pth param_key_g: params_ema strict_load_g: False resume_state: ~