Skip to content

Galileo config for pastis #204

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 14 commits into from
Oct 6, 2025
Merged

Galileo config for pastis #204

Show file tree
Hide file tree
Changes from 10 commits
Commits
Show all changes
14 commits
Select commit Hold shift + click to select a range
f4379fa
Galileo config for pastis
gabrieltseng Sep 10, 2025
e9f186f
Single yaml file
gabrieltseng Sep 11, 2025
45638b6
Fix
gabrieltseng Sep 11, 2025
895edd6
Add 'Model'
gabrieltseng Sep 11, 2025
ac2b8e5
inputs
gabrieltseng Sep 11, 2025
c062283
rslp experiment & project
gabrieltseng Sep 11, 2025
3a03a79
update image selectors
gabrieltseng Sep 11, 2025
05803d3
base
gabrieltseng Sep 11, 2025
0fd462d
nano, pass patch size
gabrieltseng Sep 11, 2025
33e46b1
patch size on init
gabrieltseng Sep 11, 2025
b1fb8eb
update ts config
gabrieltseng Oct 2, 2025
ac2ef3b
update output selector
gabrieltseng Oct 2, 2025
343ee78
fix patch size
gabrieltseng Oct 2, 2025
ac49832
Merge branch 'master' into galileo-config
gabrieltseng Oct 6, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
174 changes: 174 additions & 0 deletions data/helios/v2_pastis/basecfg_galileo_single.yaml
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -0,0 +1,174 @@
model:
class_path: rslearn.train.lightning_module.RslearnLightningModule
init_args:
model:
class_path: rslearn.models.multitask.MultiTaskModel
init_args:
encoder:
- class_path: rslearn.models.galileo.GalileoModel
init_args:
size: NANO
patch_size: 4
decoders:
segment:
- class_path: rslearn.models.unet.UNetDecoder
init_args:
in_channels: [[4, 128]]
out_channels: 20
conv_layers_per_resolution: 2
num_channels: {8: 128, 4: 128, 2: 128, 1: 128}
- class_path: rslearn.train.tasks.segmentation.SegmentationHead
lr: 0.0001
scheduler:
class_path: rslearn.train.scheduler.PlateauScheduler
init_args:
factor: 0.2
patience: 2
min_lr: 0
cooldown: 20
data:
class_path: rslearn.train.data_module.RslearnDataModule
init_args:
path: /weka/dfive-default/rslearn-eai/datasets/pastis/rslearn_dataset/
inputs:
sentinel2_0:
data_type: "raster"
layers: ["sentinel2"]
bands: ["B02", "B03", "B04", "B05", "B06", "B07", "B08", "B8A", "B11", "B12"]
passthrough: true
sentinel2_1:
data_type: "raster"
layers: ["sentinel2.1"]
bands: ["B02", "B03", "B04", "B05", "B06", "B07", "B08", "B8A", "B11", "B12"]
passthrough: true
sentinel2_2:
data_type: "raster"
layers: ["sentinel2.2"]
bands: ["B02", "B03", "B04", "B05", "B06", "B07", "B08", "B8A", "B11", "B12"]
passthrough: true
sentinel2_3:
data_type: "raster"
layers: ["sentinel2.3"]
bands: ["B02", "B03", "B04", "B05", "B06", "B07", "B08", "B8A", "B11", "B12"]
passthrough: true
sentinel2_4:
data_type: "raster"
layers: ["sentinel2.4"]
bands: ["B02", "B03", "B04", "B05", "B06", "B07", "B08", "B8A", "B11", "B12"]
passthrough: true
sentinel2_5:
data_type: "raster"
layers: ["sentinel2.5"]
bands: ["B02", "B03", "B04", "B05", "B06", "B07", "B08", "B8A", "B11", "B12"]
passthrough: true
sentinel2_6:
data_type: "raster"
layers: ["sentinel2.6"]
bands: ["B02", "B03", "B04", "B05", "B06", "B07", "B08", "B8A", "B11", "B12"]
passthrough: true
sentinel2_7:
data_type: "raster"
layers: ["sentinel2.7"]
bands: ["B02", "B03", "B04", "B05", "B06", "B07", "B08", "B8A", "B11", "B12"]
passthrough: true
sentinel2_8:
data_type: "raster"
layers: ["sentinel2.8"]
bands: ["B02", "B03", "B04", "B05", "B06", "B07", "B08", "B8A", "B11", "B12"]
passthrough: true
sentinel2_9:
data_type: "raster"
layers: ["sentinel2.9"]
bands: ["B02", "B03", "B04", "B05", "B06", "B07", "B08", "B8A", "B11", "B12"]
passthrough: true
sentinel2_10:
data_type: "raster"
layers: ["sentinel2.10"]
bands: ["B02", "B03", "B04", "B05", "B06", "B07", "B08", "B8A", "B11", "B12"]
passthrough: true
sentinel2_11:
data_type: "raster"
layers: ["sentinel2.11"]
bands: ["B02", "B03", "B04", "B05", "B06", "B07", "B08", "B8A", "B11", "B12"]
passthrough: true
targets:
data_type: "raster"
layers: ["label"]
bands: ["class"]
is_target: true
default_config:
transforms:
- class_path: rslearn.train.transforms.concatenate.Concatenate
init_args:
selections:
sentinel2_0: []
sentinel2_1: []
sentinel2_2: []
sentinel2_3: []
sentinel2_4: []
sentinel2_5: []
sentinel2_6: []
sentinel2_7: []
sentinel2_8: []
sentinel2_9: []
sentinel2_10: []
sentinel2_11: []
output_selector: s2
task:
class_path: rslearn.train.tasks.multi_task.MultiTask
init_args:
tasks:
segment:
class_path: rslearn.train.tasks.segmentation.SegmentationTask
init_args:
num_classes: 20
remap_values: [[0, 1], [0, 255]]
zero_is_invalid: true
metric_kwargs:
average: "micro"
enable_miou_metric: true
input_mapping:
segment:
targets: "targets"
train_config:
groups: ["fold1", "fold2", "fold3"]
transforms:
- class_path: rslearn.train.transforms.concatenate.Concatenate
init_args:
selections:
sentinel2_0: []
sentinel2_1: []
sentinel2_2: []
sentinel2_3: []
sentinel2_4: []
sentinel2_5: []
sentinel2_6: []
sentinel2_7: []
sentinel2_8: []
sentinel2_9: []
sentinel2_10: []
sentinel2_11: []
output_selector: s2
- class_path: rslearn.train.transforms.flip.Flip
init_args:
image_selectors: ["s2", "target/segment/classes", "target/segment/valid"]
batch_size: 8
num_workers: 16
val_config:
groups: ["fold4"]
test_config:
groups: ["fold5"]
trainer:
max_epochs: 500
callbacks:
- class_path: lightning.pytorch.callbacks.LearningRateMonitor
init_args:
logging_interval: "epoch"
- class_path: lightning.pytorch.callbacks.ModelCheckpoint
init_args:
save_top_k: 1
save_last: true
monitor: val_segment/accuracy
mode: max
rslp_project: placeholder
rslp_experiment: placeholder
124 changes: 124 additions & 0 deletions data/helios/v2_pastis/basecfg_galileo_ts.yaml
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -0,0 +1,124 @@
model:
init_args:
model:
init_args:
encoder:
- class_path: rslearn.models.galileo.Galileo
init_args:
size: TINY
decoders:
segment:
- class_path: rslearn.models.unet.UNetDecoder
init_args:
in_channels: [[8, 192]]
out_channels: 20
conv_layers_per_resolution: 2
num_channels: {8: 512, 4: 512, 2: 256, 1: 128}
- class_path: rslearn.train.tasks.segmentation.SegmentationHead
data:
init_args:
inputs:
sentinel2_0:
data_type: "raster"
layers: ["sentinel2"]
bands: ["B02", "B03", "B04", "B05", "B06", "B07", "B08", "B8A", "B11", "B12"]
passthrough: true
sentinel2_1:
data_type: "raster"
layers: ["sentinel2.1"]
bands: ["B02", "B03", "B04", "B05", "B06", "B07", "B08", "B8A", "B11", "B12"]
passthrough: true
sentinel2_2:
data_type: "raster"
layers: ["sentinel2.2"]
bands: ["B02", "B03", "B04", "B05", "B06", "B07", "B08", "B8A", "B11", "B12"]
passthrough: true
sentinel2_3:
data_type: "raster"
layers: ["sentinel2.3"]
bands: ["B02", "B03", "B04", "B05", "B06", "B07", "B08", "B8A", "B11", "B12"]
passthrough: true
sentinel2_4:
data_type: "raster"
layers: ["sentinel2.4"]
bands: ["B02", "B03", "B04", "B05", "B06", "B07", "B08", "B8A", "B11", "B12"]
passthrough: true
sentinel2_5:
data_type: "raster"
layers: ["sentinel2.5"]
bands: ["B02", "B03", "B04", "B05", "B06", "B07", "B08", "B8A", "B11", "B12"]
passthrough: true
sentinel2_6:
data_type: "raster"
layers: ["sentinel2.6"]
bands: ["B02", "B03", "B04", "B05", "B06", "B07", "B08", "B8A", "B11", "B12"]
passthrough: true
sentinel2_7:
data_type: "raster"
layers: ["sentinel2.7"]
bands: ["B02", "B03", "B04", "B05", "B06", "B07", "B08", "B8A", "B11", "B12"]
passthrough: true
sentinel2_8:
data_type: "raster"
layers: ["sentinel2.8"]
bands: ["B02", "B03", "B04", "B05", "B06", "B07", "B08", "B8A", "B11", "B12"]
passthrough: true
sentinel2_9:
data_type: "raster"
layers: ["sentinel2.9"]
bands: ["B02", "B03", "B04", "B05", "B06", "B07", "B08", "B8A", "B11", "B12"]
passthrough: true
sentinel2_10:
data_type: "raster"
layers: ["sentinel2.10"]
bands: ["B02", "B03", "B04", "B05", "B06", "B07", "B08", "B8A", "B11", "B12"]
passthrough: true
sentinel2_11:
data_type: "raster"
layers: ["sentinel2.11"]
bands: ["B02", "B03", "B04", "B05", "B06", "B07", "B08", "B8A", "B11", "B12"]
passthrough: true
targets:
data_type: "raster"
layers: ["label"]
bands: ["class"]
is_target: true
default_config:
transforms:
- class_path: rslearn.train.transforms.concatenate.Concatenate
init_args:
selections:
sentinel2_0: []
sentinel2_1: []
sentinel2_2: []
sentinel2_3: []
sentinel2_4: []
sentinel2_5: []
sentinel2_6: []
sentinel2_7: []
sentinel2_8: []
sentinel2_9: []
sentinel2_10: []
sentinel2_11: []
output_selector: sentinel2
train_config:
transforms:
- class_path: rslearn.train.transforms.concatenate.Concatenate
init_args:
selections:
sentinel2_0: []
sentinel2_1: []
sentinel2_2: []
sentinel2_3: []
sentinel2_4: []
sentinel2_5: []
sentinel2_6: []
sentinel2_7: []
sentinel2_8: []
sentinel2_9: []
sentinel2_10: []
sentinel2_11: []
output_selector: sentinel2
- class_path: rslearn.train.transforms.flip.Flip
init_args:
image_selectors: ["sentinel2", "target/segment/classes", "target/segment/valid"]
Loading