Spaces:
Running
Running
Delete configs/multi_mo_multi_task.yaml
Browse files- configs/multi_mo_multi_task.yaml +0 -156
configs/multi_mo_multi_task.yaml
DELETED
@@ -1,156 +0,0 @@
|
|
1 |
-
train_dataset:
|
2 |
-
dataset:
|
3 |
-
name: paired-image-folders-multi-task
|
4 |
-
args:
|
5 |
-
# root_path_1: ./SAM_DATA_UNIFY/Overall_Update/split_image
|
6 |
-
# root_path_1: ./SAM_DATA_UNIFY2/OVERALL/split_image
|
7 |
-
# root_path_1: ./SAM_DATA_UNIFY2/ISAID/split_image
|
8 |
-
# root_path_1: [{'ISAID': './SAM_DATA_UNIFY2/ISAID/split_image', 'WHU': './SAM_DATA_UNIFY2/WHU-OPT/split_images'}]
|
9 |
-
# root_path_1: [{'Decoder1': "/workspace/SAM_DATA_UNIFY3/Decoder1/split_image/", 'Decoder2': "/workspace/SAM_DATA_UNIFY3/Decoder2/split_image/"}]
|
10 |
-
root_path_1: [{'Decoder1': "/workspace/SAM_DATA_UNIFY4/Decoder1/image/", 'Decoder2': "/workspace/SAM_DATA_UNIFY4/Decoder2/image/"}]
|
11 |
-
# root_path_2: ./SAM_DATA_UNIFY/Overall_Update/split_gt
|
12 |
-
# root_path_2: ./SAM_DATA_UNIFY2/OVERALL/split_gt
|
13 |
-
# root_path_2: ./SAM_DATA_UNIFY2/ISAID/split_gt
|
14 |
-
# root_path_2: [{'ISAID': './SAM_DATA_UNIFY2/ISAID/split_gt', 'WHU': './SAM_DATA_UNIFY2/WHU-OPT/split_gt'}]
|
15 |
-
# root_path_2: [{'Decoder1': "/workspace/SAM_DATA_UNIFY3/Decoder1/split_gt/", 'Decoder2': "/workspace/SAM_DATA_UNIFY3/Decoder2/split_gt/"}]
|
16 |
-
root_path_2: [{'Decoder1': "/workspace/SAM_DATA_UNIFY4/Decoder1/gt/", 'Decoder2': "/workspace/SAM_DATA_UNIFY4/Decoder2/gt/"}]
|
17 |
-
cache: nones
|
18 |
-
split_key: train
|
19 |
-
wrapper:
|
20 |
-
name: train_multi_task
|
21 |
-
args:
|
22 |
-
inp_size: 1024
|
23 |
-
augment: false
|
24 |
-
# batch_size: 2
|
25 |
-
batch_size: 2
|
26 |
-
|
27 |
-
val_dataset:
|
28 |
-
dataset:
|
29 |
-
name: paired-image-folders-multi-task
|
30 |
-
args:
|
31 |
-
# root_path_1: ./SAM_DATA_UNIFY2/OVERALL/split_image
|
32 |
-
# root_path_1: [{'ISAID': './SAM_DATA_UNIFY2/ISAID/split_image', 'WHU': './SAM_DATA_UNIFY2/WHU-OPT/split_images'}]
|
33 |
-
# root_path_1: [{'Decoder1': "/workspace/SAM_DATA_UNIFY3/Decoder1/split_image/", 'Decoder2': "/workspace/SAM_DATA_UNIFY3/Decoder2/split_image/"}]
|
34 |
-
root_path_1: [{'Decoder1': "/workspace/SAM_DATA_UNIFY4/Decoder1/image/", 'Decoder2': "/workspace/SAM_DATA_UNIFY4/Decoder2/image/"}]
|
35 |
-
# root_path_2: ./SAM_DATA_UNIFY2/OVERALL/split_gt
|
36 |
-
# root_path_2: [{'ISAID': './SAM_DATA_UNIFY2/ISAID/split_gt', 'WHU': './SAM_DATA_UNIFY2/WHU-OPT/split_gt'}]
|
37 |
-
# root_path_2: [{'Decoder1': "/workspace/SAM_DATA_UNIFY3/Decoder1/split_gt/", 'Decoder2': "/workspace/SAM_DATA_UNIFY3/Decoder2/split_gt/"}]
|
38 |
-
root_path_2: [{'Decoder1': "/workspace/SAM_DATA_UNIFY4/Decoder1/gt/", 'Decoder2': "/workspace/SAM_DATA_UNIFY4/Decoder2/gt/"}]
|
39 |
-
cache: none
|
40 |
-
split_key: test
|
41 |
-
wrapper:
|
42 |
-
name: val_multi_task
|
43 |
-
args:
|
44 |
-
inp_size: 1024
|
45 |
-
# batch_size: 2
|
46 |
-
batch_size: 1
|
47 |
-
|
48 |
-
test_dataset:
|
49 |
-
dataset:
|
50 |
-
name: paired-image-folders
|
51 |
-
args:
|
52 |
-
|
53 |
-
# root_path_1: ./SAM_DATA_UNIFY3/ISAID/split_image
|
54 |
-
# root_path_1: ./SAM_DATA_UNIFY3/GANFEN/split_image
|
55 |
-
# root_path_1: ./SAM_DATA_UNIFY3/SAR2020/split_image_ov500
|
56 |
-
# root_path_1: ./SAM_DATA_UNIFY3/ISAID/split_image
|
57 |
-
# root_path_1: ./SAM_DATA_UNIFY4/SAR2020/split_image_ov500
|
58 |
-
# root_path_1: ./SAM_DATA_UNIFY4/GAOFEN/split_image
|
59 |
-
# root_path_1: ./SAM_DATA_UNIFY4/Vaihingen/image1
|
60 |
-
# root_path_1: ./SAM_DATA_UNIFY4/SAR2020/split_image_ov500
|
61 |
-
# root_path_1: ./SAM_DATA_UNIFY4/Potsdam/image1
|
62 |
-
# root_path_1: ./SAM_DATA_UNIFY4/whu-opt-sar/image_sar
|
63 |
-
root_path_1: /workspace/AIService/FoundationModel/sam_adapter_01/TwoDecoder_data/Prompt_GUOLV_Data/prompt_all1/image
|
64 |
-
|
65 |
-
# root_path_2: ./SAM_DATA_UNIFY3/ISAID/split_gt
|
66 |
-
# root_path_2: ./SAM_DATA_UNIFY3/GANFEN/gt_decoder1
|
67 |
-
# root_path_2: ./SAM_DATA_UNIFY3/GANFEN/gt_decoder2
|
68 |
-
# root_path_2: ./SAM_DATA_UNIFY3/SAR2020/gt_decoder2
|
69 |
-
# root_path_2: ./SAM_DATA_UNIFY3/ISAID/split_gt
|
70 |
-
# root_path_2: ./SAM_DATA_UNIFY4/SAR2020/gt_decoder2
|
71 |
-
# root_path_2: ./SAM_DATA_UNIFY4/GAOFEN/gt_decoder1_update
|
72 |
-
# root_path_2: ./SAM_DATA_UNIFY4/Vaihingen/gt2
|
73 |
-
# root_path_2: ./SAM_DATA_UNIFY4/Potsdam/gt1
|
74 |
-
# root_path_2: ./SAM_DATA_UNIFY4/SAR2020/gt_decoder2
|
75 |
-
root_path_2: /workspace/AIService/FoundationModel/sam_adapter_01/TwoDecoder_data/Prompt_GUOLV_Data/prompt_all1/gt
|
76 |
-
# root_path_2: ./SAM_DATA_UNIFY4/whu-opt-sar/gt_sar
|
77 |
-
cache: none
|
78 |
-
split_key: test
|
79 |
-
wrapper:
|
80 |
-
name: val
|
81 |
-
args:
|
82 |
-
# inp_size: 1024
|
83 |
-
inp_size: 1024
|
84 |
-
batch_size: 1
|
85 |
-
|
86 |
-
#eval_type: cod
|
87 |
-
eval_type: f1
|
88 |
-
#sam_checkpoint: ./pretrained/sam_vit_l_0b3195.pth
|
89 |
-
sam_checkpoint: sam_vit_h_4b8939.pth
|
90 |
-
data_norm:
|
91 |
-
inp:
|
92 |
-
sub:
|
93 |
-
- 0.5
|
94 |
-
div:
|
95 |
-
- 0.5
|
96 |
-
gt:
|
97 |
-
sub:
|
98 |
-
- 0.5
|
99 |
-
div:
|
100 |
-
- 0.5
|
101 |
-
gt_rgb:
|
102 |
-
sub:
|
103 |
-
- 0.5
|
104 |
-
div:
|
105 |
-
- 0.5
|
106 |
-
model:
|
107 |
-
name: sam
|
108 |
-
args:
|
109 |
-
inp_size: 1024
|
110 |
-
# loss: iou
|
111 |
-
loss: cr
|
112 |
-
encoder_mode:
|
113 |
-
name: sam
|
114 |
-
img_size: 1024
|
115 |
-
mlp_ratio: 4
|
116 |
-
patch_size: 16
|
117 |
-
qkv_bias: true
|
118 |
-
use_rel_pos: true
|
119 |
-
window_size: 14
|
120 |
-
out_chans: 256
|
121 |
-
scale_factor: 32
|
122 |
-
input_type: fft
|
123 |
-
freq_nums: 0.25
|
124 |
-
prompt_type: highpass
|
125 |
-
prompt_embed_dim: 256
|
126 |
-
tuning_stage: 1234
|
127 |
-
handcrafted_tune: true
|
128 |
-
embedding_tune: true
|
129 |
-
adaptor: adaptor
|
130 |
-
embed_dim: 1280
|
131 |
-
depth: 32
|
132 |
-
num_heads: 16
|
133 |
-
global_attn_indexes:
|
134 |
-
- 7
|
135 |
-
- 15
|
136 |
-
- 23
|
137 |
-
- 31
|
138 |
-
optimizer:
|
139 |
-
name: adamw
|
140 |
-
args:
|
141 |
-
# lr: 0.0002
|
142 |
-
# lr: 0.00002
|
143 |
-
lr: 0.00008
|
144 |
-
lr_min: 1.0e-8
|
145 |
-
#epoch_max: 20
|
146 |
-
epoch_max: 100
|
147 |
-
|
148 |
-
multi_step_lr:
|
149 |
-
milestones:
|
150 |
-
- 1
|
151 |
-
gamma: 0.1
|
152 |
-
epoch_val: 100
|
153 |
-
epoch_save: 1
|
154 |
-
|
155 |
-
#resume: 60
|
156 |
-
#start_epoch: 60
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|