build: bump ccrestoration to v0.2.1

Author: routineLife1

pyproject.toml

@@ -44,7 +44,7 @@ version = "3.0.2"
 # Requirements
 [tool.poetry.dependencies]
 PyYAML = "*"
-ccrestoration = "0.1.2"
+ccrestoration = "0.2.1"
 loguru = "*"
 pydantic = "*"
 python = "^3.9"

scripts/ModelOptions.ts

@@ -18,6 +18,33 @@ export const modelOptions: any[] = [
   { label: 'RealESRGAN_Ani4Kv2_Compact_2x', value: 'RealESRGAN_Ani4Kv2_Compact_2x.pth' },
   { label: 'RealESRGAN_APISR_RRDB_GAN_generator_2x', value: 'RealESRGAN_APISR_RRDB_GAN_generator_2x.pth' },
   { label: 'RealESRGAN_APISR_RRDB_GAN_generator_4x', value: 'RealESRGAN_APISR_RRDB_GAN_generator_4x.pth' },
+  { label: 'DAT_S_2x', value: 'DAT_S_2x.pth' },
+  { label: 'DAT_S_3x', value: 'DAT_S_3x.pth' },
+  { label: 'DAT_S_4x', value: 'DAT_S_4x.pth' },
+  { label: 'DAT_2x', value: 'DAT_2x.pth' },
+  { label: 'DAT_3x', value: 'DAT_3x.pth' },
+  { label: 'DAT_4x', value: 'DAT_4x.pth' },
+  { label: 'DAT_2_2x', value: 'DAT_2_2x.pth' },
+  { label: 'DAT_2_3x', value: 'DAT_2_3x.pth' },
+  { label: 'DAT_2_4x', value: 'DAT_2_4x.pth' },
+  { label: 'DAT_light_2x', value: 'DAT_light_2x.pth' },
+  { label: 'DAT_light_3x', value: 'DAT_light_3x.pth' },
+  { label: 'DAT_light_4x', value: 'DAT_light_4x.pth' },
+  { label: 'DAT_APISR_GAN_generator_4x', value: 'DAT_APISR_GAN_generator_4x.pth' },
+  { label: 'HAT_S_2x', value: 'HAT_S_2x.pth' },
+  { label: 'HAT_S_3x', value: 'HAT_S_3x.pth' },
+  { label: 'HAT_S_4x', value: 'HAT_S_4x.pth' },
+  { label: 'HAT_2x', value: 'HAT_2x.pth' },
+  { label: 'HAT_3x', value: 'HAT_3x.pth' },
+  { label: 'HAT_4x', value: 'HAT_4x.pth' },
+  { label: 'HAT_Real_GAN_sharper_4x', value: 'HAT_Real_GAN_sharper_4x.pth' },
+  { label: 'HAT_Real_GAN_4x', value: 'HAT_Real_GAN_4x.pth' },
+  { label: 'HAT_ImageNet_pretrain_2x', value: 'HAT_ImageNet_pretrain_2x.pth' },
+  { label: 'HAT_ImageNet_pretrain_3x', value: 'HAT_ImageNet_pretrain_3x.pth' },
+  { label: 'HAT_ImageNet_pretrain_4x', value: 'HAT_ImageNet_pretrain_4x.pth' },
+  { label: 'HAT_L_ImageNet_pretrain_2x', value: 'HAT_L_ImageNet_pretrain_2x.pth' },
+  { label: 'HAT_L_ImageNet_pretrain_3x', value: 'HAT_L_ImageNet_pretrain_3x.pth' },
+  { label: 'HAT_L_ImageNet_pretrain_4x', value: 'HAT_L_ImageNet_pretrain_4x.pth' },
   { label: 'RealCUGAN_Conservative_2x', value: 'RealCUGAN_Conservative_2x.pth' },
   { label: 'RealCUGAN_Denoise1x_2x', value: 'RealCUGAN_Denoise1x_2x.pth' },
   { label: 'RealCUGAN_Denoise2x_2x', value: 'RealCUGAN_Denoise2x_2x.pth' },
@@ -47,17 +74,7 @@ export const modelOptions: any[] = [
   { label: 'SCUNet_color_50_1x', value: 'SCUNet_color_50_1x.pth' },
   { label: 'SCUNet_color_real_psnr_1x', value: 'SCUNet_color_real_psnr_1x.pth' },
   { label: 'SCUNet_color_real_gan_1x', value: 'SCUNet_color_real_gan_1x.pth' },
-  { label: 'DAT_S_2x', value: 'DAT_S_2x.pth' },
-  { label: 'DAT_S_3x', value: 'DAT_S_3x.pth' },
-  { label: 'DAT_S_4x', value: 'DAT_S_4x.pth' },
-  { label: 'DAT_2x', value: 'DAT_2x.pth' },
-  { label: 'DAT_3x', value: 'DAT_3x.pth' },
-  { label: 'DAT_4x', value: 'DAT_4x.pth' },
-  { label: 'DAT_2_2x', value: 'DAT_2_2x.pth' },
-  { label: 'DAT_2_3x', value: 'DAT_2_3x.pth' },
-  { label: 'DAT_2_4x', value: 'DAT_2_4x.pth' },
-  { label: 'DAT_light_2x', value: 'DAT_light_2x.pth' },
-  { label: 'DAT_light_3x', value: 'DAT_light_3x.pth' },
-  { label: 'DAT_light_4x', value: 'DAT_light_4x.pth' },
-  { label: 'DAT_APISR_GAN_generator_4x', value: 'DAT_APISR_GAN_generator_4x.pth' },
+  { label: 'SRCNN_2x', value: 'SRCNN_2x.pth' },
+  { label: 'SRCNN_3x', value: 'SRCNN_3x.pth' },
+  { label: 'SRCNN_4x', value: 'SRCNN_4x.pth' },
 ]

scripts/gen_pretrained_model_name.py

@@ -30,15 +30,17 @@ def gen_ts() -> None:
     # add new sr models here
     all_sr_models = [
         ModelType.RealESRGAN,
+        ModelType.DAT,
+        ModelType.HAT,
         ModelType.RealCUGAN,
         ModelType.EDSR,
         ModelType.SwinIR,
         ModelType.SCUNet,
-        ModelType.DAT,
+        ModelType.SRCNN,
     ]
 
-    for cfg in ConfigType:
-        for m in all_sr_models:
+    for m in all_sr_models:
+        for cfg in ConfigType:
             if cfg.startswith(m):
                 config_list.append(cfg)  # type: ignore