Use SAM_Perceptual(train=True, weights=[0,1,1,0]) as training loss instead of IMPACTReg

CBCT/Config.yml

@@ -18,21 +18,9 @@ Trainer:
                   stop: None
                   accumulation: false
                   reduction: mean
-                IMPACTReg/1:
-                  name: SAM
-                  model_name: SAM2.1/SAM2.1_Small.pt
-                  shape:
-                  - 0
-                  - 0
-                  in_channels: 3
-                  loss: torch:nn:L1Loss
-                  weights:
-                  - 0
-                  - 1
-                  - 1
-                  size_average: None
-                  reduce: None
-                  reduction: mean
+                SAM_Perceptual:
+                  train: true
+                  weights: [0, 1, 1, 0]
                   schedulers:
                     Constant:
                       nb_step: 0
@@ -178,4 +166,4 @@ Trainer:
     monitor: []
     patience: 30
     min_delta: 0.0
-    mode: min
+    mode: min

Finetune/Config.yml

@@ -18,21 +18,9 @@ Trainer:
                   stop: None
                   accumulation: false
                   reduction: mean
-                IMPACTReg/1:
-                  name: SAM
-                  model_name: SAM2.1/SAM2.1_Small.pt
-                  shape:
-                  - 0
-                  - 0
-                  in_channels: 3
-                  loss: torch:nn:L1Loss
-                  weights:
-                  - 0
-                  - 1
-                  - 1
-                  size_average: None
-                  reduce: None
-                  reduction: mean
+                SAM_Perceptual:
+                  train: true
+                  weights: [0, 1, 1, 0]
                   schedulers:
                     Constant:
                       nb_step: 0
@@ -178,4 +166,4 @@ Trainer:
     monitor: []
     patience: 30
     min_delta: 0.0
-    mode: min
+    mode: min

MR/Config.yml

@@ -18,21 +18,9 @@ Trainer:
                   stop: None
                   accumulation: false
                   reduction: mean
-                IMPACTReg/1:
-                  name: SAM
-                  model_name: SAM2.1/SAM2.1_Small.pt
-                  shape:
-                  - 0
-                  - 0
-                  in_channels: 3
-                  loss: torch:nn:L1Loss
-                  weights:
-                  - 0
-                  - 1
-                  - 1
-                  size_average: None
-                  reduce: None
-                  reduction: mean
+                SAM_Perceptual:
+                  train: true
+                  weights: [0, 1, 1, 0]
                   schedulers:
                     Constant:
                       nb_step: 0
@@ -178,4 +166,4 @@ Trainer:
     monitor: []
     patience: 30
     min_delta: 0.0
-    mode: min
+    mode: min

MR_CBCT/Config.yml

@@ -18,21 +18,9 @@ Trainer:
                   stop: None
                   accumulation: false
                   reduction: mean
-                IMPACTReg/1:
-                  name: SAM
-                  model_name: SAM2.1/SAM2.1_Small.pt
-                  shape:
-                  - 0
-                  - 0
-                  in_channels: 3
-                  loss: torch:nn:L1Loss
-                  weights:
-                  - 0
-                  - 1
-                  - 1
-                  size_average: None
-                  reduce: None
-                  reduction: mean
+                SAM_Perceptual:
+                  train: true
+                  weights: [0, 1, 1, 0]
                   schedulers:
                     Constant:
                       nb_step: 0
@@ -178,4 +166,4 @@ Trainer:
     monitor: []
     patience: 30
     min_delta: 0.0
-    mode: min
+    mode: min