First commit.

cce6e1bf · chenych · cce6e1bf · cce6e1bf · cce6e1bf · cce6e1bf
Commit cce6e1bf authored Nov 21, 2023 by chenych
20 changed files
--- a/adet/modeling/vitae_v2/token_transformer.py
+++ b/adet/modeling/vitae_v2/token_transformer.py
--- a/adet/modeling/vitae_v2/vitae_v2.py
+++ b/adet/modeling/vitae_v2/vitae_v2.py
--- a/adet/modeling/vitae_v2/window.py
+++ b/adet/modeling/vitae_v2/window.py
--- a/adet/utils/__init__.py
+++ b/adet/utils/__init__.py
--- a/adet/utils/comm.py
+++ b/adet/utils/comm.py
--- a/adet/utils/curve_utils.py
+++ b/adet/utils/curve_utils.py
--- a/adet/utils/misc.py
+++ b/adet/utils/misc.py
--- a/adet/utils/visualizer.py
+++ b/adet/utils/visualizer.py
--- a/configs/R_101/Base_det.yaml
+++ b/configs/R_101/Base_det.yaml
--- a/configs/R_101/TotalText/finetune_150k_tt_mlt_13_15.yaml
+++ b/configs/R_101/TotalText/finetune_150k_tt_mlt_13_15.yaml
+_BASE_: "../Base_det.yaml"
+
+MODEL:
+  WEIGHTS: "output/R101/150k_tt_mlt_13_15/pretrain/model_final.pth"
+
+DATASETS:
+  TRAIN: ("totaltext_train",)
+  TEST: ("totaltext_test",)
+
+SOLVER:
+  IMS_PER_BATCH: 8
+  BASE_LR: 1e-5
+  LR_BACKBONE: 1e-6
+  WARMUP_ITERS: 0
+  STEPS: (100000,)  # no step
+  MAX_ITER: 10000
+  CHECKPOINT_PERIOD: 2000
+
+TEST:
+  EVAL_PERIOD: 1000
+
+OUTPUT_DIR: "output/R101/150k_tt_mlt_13_15/finetune/totaltext"
\ No newline at end of file
--- a/configs/R_101/pretrain/150k_tt_mlt_13_15.yaml
+++ b/configs/R_101/pretrain/150k_tt_mlt_13_15.yaml
--- a/configs/R_50/Base_det.yaml
+++ b/configs/R_50/Base_det.yaml
--- a/configs/R_50/CTW1500/finetune_96voc_50maxlen.yaml
+++ b/configs/R_50/CTW1500/finetune_96voc_50maxlen.yaml
--- a/configs/R_50/CTW1500/pretrain_96voc_50maxlen.yaml
+++ b/configs/R_50/CTW1500/pretrain_96voc_50maxlen.yaml
--- a/configs/R_50/IC15/finetune_150k_tt_mlt_13_15.yaml
+++ b/configs/R_50/IC15/finetune_150k_tt_mlt_13_15.yaml
--- a/configs/R_50/IC15/finetune_150k_tt_mlt_13_15_textocr.yaml
+++ b/configs/R_50/IC15/finetune_150k_tt_mlt_13_15_textocr.yaml
--- a/configs/R_50/ReCTS/finetune.yaml
+++ b/configs/R_50/ReCTS/finetune.yaml
--- a/configs/R_50/ReCTS/pretrain.yaml
+++ b/configs/R_50/ReCTS/pretrain.yaml
--- a/configs/R_50/TotalText/finetune_150k_tt.yaml
+++ b/configs/R_50/TotalText/finetune_150k_tt.yaml
--- a/configs/R_50/TotalText/finetune_150k_tt_mlt_13_15.yaml
+++ b/configs/R_50/TotalText/finetune_150k_tt_mlt_13_15.yaml