Преглед на файлове

Update experiment scripts

Bobholamovic преди 3 години
родител
ревизия
87596b5f1f
променени са 19 файла, в които са добавени 58 реда и са изтрити 70 реда
  1. 0 6
      examples/rs_research/configs/levircd/bit.yaml
  2. 8 0
      examples/rs_research/configs/levircd/custom_model/custom_model_c.yaml
  3. 8 0
      examples/rs_research/configs/levircd/custom_model/custom_model_cs.yaml
  4. 8 0
      examples/rs_research/configs/levircd/custom_model/custom_model_cst.yaml
  5. 8 0
      examples/rs_research/configs/levircd/custom_model/custom_model_ct.yaml
  6. 8 0
      examples/rs_research/configs/levircd/custom_model/custom_model_s.yaml
  7. 8 0
      examples/rs_research/configs/levircd/custom_model/custom_model_st.yaml
  8. 8 0
      examples/rs_research/configs/levircd/custom_model/custom_model_t.yaml
  9. 0 11
      examples/rs_research/configs/levircd/custom_model/iterative_bit_iter2.yaml
  10. 0 11
      examples/rs_research/configs/levircd/custom_model/iterative_bit_iter3.yaml
  11. 0 11
      examples/rs_research/configs/levircd/custom_model/iterative_bit_iter4.yaml
  12. 0 2
      examples/rs_research/configs/levircd/fc_ef.yaml
  13. 0 2
      examples/rs_research/configs/levircd/fc_siam_conc.yaml
  14. 0 2
      examples/rs_research/configs/levircd/fc_siam_diff.yaml
  15. 0 6
      examples/rs_research/configs/levircd/stanet.yaml
  16. 0 6
      examples/rs_research/configs/svcd/bit.yaml
  17. 1 6
      examples/rs_research/configs/svcd/custom_model.yaml
  18. 0 6
      examples/rs_research/configs/svcd/stanet.yaml
  19. 1 1
      examples/rs_research/scripts/run_ablation.sh

+ 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}"