sincostanx
commited on
Commit
•
4ffa655
1
Parent(s):
a237bfa
add models and config files
Browse files- MSS_resnet50.pt +3 -0
- MSS_resnet50.yaml +62 -0
- OLITIRS_resnet50.pt +3 -0
- OLITIRS_resnet50.yml +62 -0
- TM_resnet50.pt +3 -0
- TM_resnet50.yaml +62 -0
MSS_resnet50.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:fa3f84a34abbfe215c7f835d5661ed3fe0ee3f0484c28c33d84e26a8540f197a
|
3 |
+
size 390870803
|
MSS_resnet50.yaml
ADDED
@@ -0,0 +1,62 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
architecture: "Unet"
|
2 |
+
|
3 |
+
model_params:
|
4 |
+
encoder_name: "resnet50"
|
5 |
+
encoder_weights: "swsl"
|
6 |
+
in_channels: 5
|
7 |
+
classes: 11
|
8 |
+
|
9 |
+
device: "cuda"
|
10 |
+
seed: 0
|
11 |
+
save_path: "./checkpoints/"
|
12 |
+
|
13 |
+
training:
|
14 |
+
dataloader_params:
|
15 |
+
landsat:
|
16 |
+
root: "./Megacity2035_MSS/train"
|
17 |
+
filename_glob: "**/*.tif"
|
18 |
+
exclude: null
|
19 |
+
esa:
|
20 |
+
root: "./Megacity2035_pseudolabel/train"
|
21 |
+
filename_glob: "**/*.tif"
|
22 |
+
exclude: null
|
23 |
+
sensor_type: "MSS"
|
24 |
+
tile_per_loader: 16
|
25 |
+
length: 50000
|
26 |
+
batch_size: 128
|
27 |
+
num_workers: 8
|
28 |
+
|
29 |
+
trainer_params:
|
30 |
+
epoch: 100
|
31 |
+
gradient_accumulation_factor: null
|
32 |
+
|
33 |
+
loss_fn_params:
|
34 |
+
weights: [1.0, 1.0]
|
35 |
+
FocalLoss:
|
36 |
+
mode: "multiclass"
|
37 |
+
alpha: null
|
38 |
+
gamma: 2.0
|
39 |
+
normalized: False
|
40 |
+
TverskyLoss:
|
41 |
+
mode: "multiclass"
|
42 |
+
alpha: 0.3
|
43 |
+
beta: 0.7
|
44 |
+
gamma: 2.0
|
45 |
+
|
46 |
+
optimizer_params:
|
47 |
+
lr: 5.0e-4
|
48 |
+
weight_decay: 1.0e-5
|
49 |
+
|
50 |
+
scheduler_params:
|
51 |
+
step_size: 30
|
52 |
+
gamma: 0.5
|
53 |
+
last_epoch: -1
|
54 |
+
|
55 |
+
validate_params:
|
56 |
+
root: "./"
|
57 |
+
img_glob: "Megacity2035_MSS/val/**/*.tif"
|
58 |
+
gt_glob: "Megacity2035_pseudolabel/val/**/*.tif"
|
59 |
+
|
60 |
+
logger_params:
|
61 |
+
PROJECT: "GUC_master"
|
62 |
+
NAME: "dummy_landsat1"
|
OLITIRS_resnet50.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:6ef3dea7727176237f0f2979705f09263b5b78c2e7c58fa2f4f930c5264fb39e
|
3 |
+
size 391088000
|
OLITIRS_resnet50.yml
ADDED
@@ -0,0 +1,62 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
architecture: "Unet"
|
2 |
+
|
3 |
+
model_params:
|
4 |
+
encoder_name: "resnet50"
|
5 |
+
encoder_weights: "swsl"
|
6 |
+
in_channels: 11
|
7 |
+
classes: 11
|
8 |
+
|
9 |
+
device: "cuda"
|
10 |
+
seed: 0
|
11 |
+
save_path: "./checkpoints/"
|
12 |
+
|
13 |
+
training:
|
14 |
+
dataloader_params:
|
15 |
+
landsat:
|
16 |
+
root: "./Megacity2035/train"
|
17 |
+
filename_glob: "**/landsat8_2021.tif"
|
18 |
+
exclude: null
|
19 |
+
esa:
|
20 |
+
root: "./Megacity2035/train"
|
21 |
+
filename_glob: "**/ESAv200_.tif"
|
22 |
+
exclude: null
|
23 |
+
sensor_type: "OLITIRS"
|
24 |
+
tile_per_loader: 24
|
25 |
+
length: 50000
|
26 |
+
batch_size: 128
|
27 |
+
num_workers: 8
|
28 |
+
|
29 |
+
trainer_params:
|
30 |
+
epoch: 100
|
31 |
+
gradient_accumulation_factor: null
|
32 |
+
|
33 |
+
loss_fn_params:
|
34 |
+
weights: [1.0, 1.0]
|
35 |
+
FocalLoss:
|
36 |
+
mode: "multiclass"
|
37 |
+
alpha: null
|
38 |
+
gamma: 2.0
|
39 |
+
normalized: False
|
40 |
+
TverskyLoss:
|
41 |
+
mode: "multiclass"
|
42 |
+
alpha: 0.3
|
43 |
+
beta: 0.7
|
44 |
+
gamma: 2.0
|
45 |
+
|
46 |
+
optimizer_params:
|
47 |
+
lr: 5.0e-4
|
48 |
+
weight_decay: 1.0e-5
|
49 |
+
|
50 |
+
scheduler_params:
|
51 |
+
step_size: 30
|
52 |
+
gamma: 0.5
|
53 |
+
last_epoch: -1
|
54 |
+
|
55 |
+
validate_params:
|
56 |
+
root: "./Megacity2035/val"
|
57 |
+
img_glob: "**/landsat8_2021.tif"
|
58 |
+
gt_glob: "**/ESAv200_.tif"
|
59 |
+
|
60 |
+
logger_params:
|
61 |
+
PROJECT: "GUC_master"
|
62 |
+
NAME: "dummy"
|
TM_resnet50.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:5791a8cf1188dd551a71b3dd6538b23f2ce131abc0495988e1e9148782583633
|
3 |
+
size 391079274
|
TM_resnet50.yaml
ADDED
@@ -0,0 +1,62 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
architecture: "Unet"
|
2 |
+
|
3 |
+
model_params:
|
4 |
+
encoder_name: "resnet50"
|
5 |
+
encoder_weights: "swsl"
|
6 |
+
in_channels: 10
|
7 |
+
classes: 11
|
8 |
+
|
9 |
+
device: "cuda"
|
10 |
+
seed: 0
|
11 |
+
save_path: "./checkpoints/"
|
12 |
+
|
13 |
+
training:
|
14 |
+
dataloader_params:
|
15 |
+
landsat:
|
16 |
+
root: "./Megacity2035/train"
|
17 |
+
filename_glob: "**/landsat7_2021.tif"
|
18 |
+
exclude: null
|
19 |
+
esa:
|
20 |
+
root: "./Megacity2035/train"
|
21 |
+
filename_glob: "**/ESAv200_.tif"
|
22 |
+
exclude: null
|
23 |
+
sensor_type: "TM"
|
24 |
+
tile_per_loader: 24
|
25 |
+
length: 50000
|
26 |
+
batch_size: 128
|
27 |
+
num_workers: 8
|
28 |
+
|
29 |
+
trainer_params:
|
30 |
+
epoch: 100
|
31 |
+
gradient_accumulation_factor: null
|
32 |
+
|
33 |
+
loss_fn_params:
|
34 |
+
weights: [1.0, 1.0]
|
35 |
+
FocalLoss:
|
36 |
+
mode: "multiclass"
|
37 |
+
alpha: null
|
38 |
+
gamma: 2.0
|
39 |
+
normalized: False
|
40 |
+
TverskyLoss:
|
41 |
+
mode: "multiclass"
|
42 |
+
alpha: 0.3
|
43 |
+
beta: 0.7
|
44 |
+
gamma: 2.0
|
45 |
+
|
46 |
+
optimizer_params:
|
47 |
+
lr: 5.0e-4
|
48 |
+
weight_decay: 1.0e-5
|
49 |
+
|
50 |
+
scheduler_params:
|
51 |
+
step_size: 30
|
52 |
+
gamma: 0.5
|
53 |
+
last_epoch: -1
|
54 |
+
|
55 |
+
validate_params:
|
56 |
+
root: "./Megacity2035/val"
|
57 |
+
img_glob: "**/landsat7_2021.tif"
|
58 |
+
gt_glob: "**/ESAv200_.tif"
|
59 |
+
|
60 |
+
logger_params:
|
61 |
+
PROJECT: "GUC_master"
|
62 |
+
NAME: "dummy_landsat7_corrected-normalized"
|