add MAT model

2022-08-22 23:24:02 +08:00
parent a5e840765e
commit 6d2b24ed6b
8 changed files with 2132 additions and 9 deletions
--- a/lama_cleaner/app/src/store/Atoms.tsx
+++ b/lama_cleaner/app/src/store/Atoms.tsx
@@ -7,6 +7,7 @@ export enum AIModel {
  LAMA = 'lama',
  LDM = 'ldm',
  ZITS = 'zits',
+  MAT = 'mat',
 }

 export const fileState = atom<File | undefined>({
@@ -80,6 +81,12 @@ const defaultHDSettings: ModelsHDSettings = {
    hdStrategyCropTrigerSize: 1024,
    hdStrategyCropMargin: 128,
  },
+  [AIModel.MAT]: {
+    hdStrategy: HDStrategy.CROP,
+    hdStrategyResizeLimit: 1024,
+    hdStrategyCropTrigerSize: 512,
+    hdStrategyCropMargin: 128,
+  },
 }

 export const settingStateDefault: Settings = {