feat: support HAT && SRCNN (#607)

Tohrusky · Dec 14, 2024 · 70b9f00 · 70b9f00
1 parent 674dec3
commit 70b9f00
Show file tree

Hide file tree

Showing 2 changed files with 33 additions and 16 deletions.
diff --git a/resources/download-core.js b/resources/download-core.js
@@ -9,11 +9,11 @@ const child_process = require('child_process')
 
 const coreDict = {
   'macos-arm64':
-    'https://github.com/Tohrusky/Final2x-core/releases/download/2024-11-09/Final2x-core-macos-arm64.7z',
+    'https://github.com/Tohrusky/Final2x-core/releases/download/2024-12-14/Final2x-core-macos-arm64.7z',
   'macos-x64':
-    'https://github.com/Tohrusky/Final2x-core/releases/download/2024-11-09/Final2x-core-macos-x64.7z',
+    'https://github.com/Tohrusky/Final2x-core/releases/download/2024-12-14/Final2x-core-macos-x64.7z',
   'windows-x64':
-    'https://github.com/Tohrusky/Final2x-core/releases/download/2024-11-09/Final2x-core-windows-latest.7z'
+    'https://github.com/Tohrusky/Final2x-core/releases/download/2024-12-14/Final2x-core-windows-latest.7z'
 }
 
 console.log('-'.repeat(50))

diff --git a/src/renderer/src/utils/ModelOptions.ts b/src/renderer/src/utils/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' },
 ]