feat: support DAT (#601)

Tohrusky · Nov 9, 2024 · bb91a8b · bb91a8b
1 parent c7ef1d2
commit bb91a8b
Show file tree

Hide file tree

Showing 3 changed files with 17 additions and 4 deletions.
diff --git a/README.md b/README.md
@@ -77,7 +77,7 @@ apt install -y libomp5 xdg-utils
 
 The following references were referenced in the development of this project:
 
-- [Final2x-core](https://github.com/Final2x/Final2x-core)
+- [Final2x-core](https://github.com/Tohrusky/Final2x-core)
 - [ccrestoration](https://github.com/TensoRaws/ccrestoration)
 - [PyTorch](https://github.com/pytorch/pytorch)
 - [ncnn](https://github.com/Tencent/ncnn)

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-08/Final2x-core-macos-arm64.7z',
+    'https://github.com/Tohrusky/Final2x-core/releases/download/2024-11-09/Final2x-core-macos-arm64.7z',
   'macos-x64':
-    'https://github.com/Tohrusky/Final2x-core/releases/download/2024-11-08/Final2x-core-macos-x64.7z',
+    'https://github.com/Tohrusky/Final2x-core/releases/download/2024-11-09/Final2x-core-macos-x64.7z',
   'windows-x64':
-    'https://github.com/Tohrusky/Final2x-core/releases/download/2024-11-08/Final2x-core-windows-latest.7z'
+    'https://github.com/Tohrusky/Final2x-core/releases/download/2024-11-09/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
@@ -47,4 +47,17 @@ 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' },
 ]