Prechádzať zdrojové kódy

Update experiment scripts

Bobholamovic 2 rokov pred
rodič
commit
87596b5f1f

+ 0 - 6
examples/rs_research/configs/levircd/bit.yaml

@@ -1,6 +0,0 @@
-_base_: ./levircd.yaml
-
-save_dir: ./exp/levircd/bit/
-
-model: !Node
-    type: BIT

+ 8 - 0
examples/rs_research/configs/levircd/custom_model/custom_model_c.yaml

@@ -0,0 +1,8 @@
+_base_: ../levircd.yaml
+
+save_dir: ./exp/levircd/custom_model/att_c/
+
+model: !Node
+    type: CustomTrainer
+    args:
+        att_types: c

+ 8 - 0
examples/rs_research/configs/levircd/custom_model/custom_model_cs.yaml

@@ -0,0 +1,8 @@
+_base_: ../levircd.yaml
+
+save_dir: ./exp/levircd/custom_model/att_cs/
+
+model: !Node
+    type: CustomTrainer
+    args:
+        att_types: cs

+ 8 - 0
examples/rs_research/configs/levircd/custom_model/custom_model_cst.yaml

@@ -0,0 +1,8 @@
+_base_: ../levircd.yaml
+
+save_dir: ./exp/levircd/custom_model/att_cst/
+
+model: !Node
+    type: CustomTrainer
+    args:
+        att_types: cst

+ 8 - 0
examples/rs_research/configs/levircd/custom_model/custom_model_ct.yaml

@@ -0,0 +1,8 @@
+_base_: ../levircd.yaml
+
+save_dir: ./exp/levircd/custom_model/att_ct/
+
+model: !Node
+    type: CustomTrainer
+    args:
+        att_types: ct

+ 8 - 0
examples/rs_research/configs/levircd/custom_model/custom_model_s.yaml

@@ -0,0 +1,8 @@
+_base_: ../levircd.yaml
+
+save_dir: ./exp/levircd/custom_model/att_s/
+
+model: !Node
+    type: CustomTrainer
+    args:
+        att_types: s

+ 8 - 0
examples/rs_research/configs/levircd/custom_model/custom_model_st.yaml

@@ -0,0 +1,8 @@
+_base_: ../levircd.yaml
+
+save_dir: ./exp/levircd/custom_model/att_st/
+
+model: !Node
+    type: CustomTrainer
+    args:
+        att_types: st

+ 8 - 0
examples/rs_research/configs/levircd/custom_model/custom_model_t.yaml

@@ -0,0 +1,8 @@
+_base_: ../levircd.yaml
+
+save_dir: ./exp/levircd/custom_model/att_t/
+
+model: !Node
+    type: CustomTrainer
+    args:
+        att_types: t

+ 0 - 11
examples/rs_research/configs/levircd/custom_model/iterative_bit_iter2.yaml

@@ -1,11 +0,0 @@
-_base_: ../levircd.yaml
-
-save_dir: ./exp/levircd/custom_model/iter2/
-
-model: !Node
-    type: IterativeBIT
-    args:
-        num_iters: 2
-        num_classes: 2
-        bit_kwargs:
-            in_channels: 3

+ 0 - 11
examples/rs_research/configs/levircd/custom_model/iterative_bit_iter3.yaml

@@ -1,11 +0,0 @@
-_base_: ../levircd.yaml
-
-save_dir: ./exp/levircd/custom_model/iter3/
-
-model: !Node
-    type: IterativeBIT
-    args:
-        num_iters: 3
-        num_classes: 2
-        bit_kwargs:
-            in_channels: 3

+ 0 - 11
examples/rs_research/configs/levircd/custom_model/iterative_bit_iter4.yaml

@@ -1,11 +0,0 @@
-_base_: ../levircd.yaml
-
-save_dir: ./exp/levircd/custom_model/iter4/
-
-model: !Node
-    type: IterativeBIT
-    args:
-        num_iters: 4
-        num_classes: 2
-        bit_kwargs:
-            in_channels: 3

+ 0 - 2
examples/rs_research/configs/levircd/fc_ef.yaml

@@ -4,5 +4,3 @@ save_dir: ./exp/levircd/fc_ef/
 
 model: !Node
     type: FCEarlyFusion
-    args:
-        use_dropout: True

+ 0 - 2
examples/rs_research/configs/levircd/fc_siam_conc.yaml

@@ -4,5 +4,3 @@ save_dir: ./exp/levircd/fc_siam_conc/
 
 model: !Node
     type: FCSiamConc
-    args:
-        use_dropout: True

+ 0 - 2
examples/rs_research/configs/levircd/fc_siam_diff.yaml

@@ -4,5 +4,3 @@ save_dir: ./exp/levircd/fc_siam_diff/
 
 model: !Node
     type: FCSiamDiff
-    args:
-        use_dropout: True

+ 0 - 6
examples/rs_research/configs/levircd/stanet.yaml

@@ -1,6 +0,0 @@
-_base_: ./levircd.yaml
-
-save_dir: ./exp/levircd/stanet/
-
-model: !Node
-    type: STANet

+ 0 - 6
examples/rs_research/configs/svcd/bit.yaml

@@ -1,6 +0,0 @@
-_base_: ./svcd.yaml
-
-save_dir: ./exp/svcd/bit/
-
-model: !Node
-    type: BIT

+ 1 - 6
examples/rs_research/configs/svcd/custom_model.yaml

@@ -3,9 +3,4 @@ _base_: ./svcd.yaml
 save_dir: ./exp/svcd/custom_model/
 
 model: !Node
-    type: IterativeBIT
-    args:
-        num_iters: 2
-        num_classes: 2
-        bit_kwargs:
-            in_channels: 3
+    type: CustomTrainer

+ 0 - 6
examples/rs_research/configs/svcd/stanet.yaml

@@ -1,6 +0,0 @@
-_base_: ./svcd.yaml
-
-save_dir: ./exp/svcd/stanet/
-
-model: !Node
-    type: STANet

+ 1 - 1
examples/rs_research/scripts/run_parameter_analysis.sh → examples/rs_research/scripts/run_ablation.sh

@@ -3,7 +3,7 @@
 set -e 
 
 CONFIG_DIR='configs/levircd/custom_model'
-LOG_DIR='exp/logs/parameter_analysis'
+LOG_DIR='exp/logs/ablation'
 
 mkdir -p "${LOG_DIR}"