|
|
|
seed_everything: 2 |
|
trainer: |
|
logger: true |
|
max_epochs: 100 |
|
log_every_n_steps: 1 |
|
callbacks: |
|
- class_path: EarlyStopping |
|
init_args: |
|
monitor: val/loss |
|
patience: 15 |
|
- class_path: LearningRateMonitor |
|
init_args: |
|
logging_interval: epoch |
|
enable_progress_bar: false |
|
precision: bf16-mixed |
|
|
|
model: |
|
class_path: terratorch.tasks.SemanticSegmentationTask |
|
init_args: |
|
model_factory: EncoderDecoderFactory |
|
model_args: |
|
backbone: prithvi_eo_v2_300 |
|
backbone_pretrained: true |
|
backbone_bands: ["BLUE", "GREEN", "RED", "NIR_NARROW", "SWIR_1", "SWIR_2"] |
|
necks: |
|
- name: SelectIndices |
|
indices: [5, 11, 17, 23] |
|
- name: ReshapeTokensToImage |
|
- name: LearnedInterpolateToPyramidal |
|
decoder: UNetDecoder |
|
decoder_channels: [512, 256, 128, 64] |
|
num_classes: 2 |
|
loss: ce |
|
ignore_index: -1 |
|
freeze_backbone: false |
|
plot_on_val: false |
|
class_names: [Not burned, Burn scar] |
|
|
|
optimizer: |
|
class_path: torch.optim.AdamW |
|
init_args: |
|
lr: 1.e-4 |
|
lr_scheduler: |
|
class_path: ReduceLROnPlateau |
|
init_args: |
|
monitor: val/loss |
|
factor: 0.5 |
|
patience: 4 |
|
|
|
data: |
|
class_path: GenericNonGeoSegmentationDataModule |
|
init_args: |
|
batch_size: 8 |
|
num_workers: 8 |
|
dataset_bands: |
|
- BLUE |
|
- GREEN |
|
- RED |
|
- NIR_NARROW |
|
- SWIR_1 |
|
- SWIR_2 |
|
output_bands: |
|
- BLUE |
|
- GREEN |
|
- RED |
|
- NIR_NARROW |
|
- SWIR_1 |
|
- SWIR_2 |
|
rgb_indices: |
|
- 2 |
|
- 1 |
|
- 0 |
|
train_data_root: hls_burn_scars/data |
|
val_data_root: hls_burn_scars/data |
|
test_data_root: hls_burn_scars/data |
|
train_split: hls_burn_scars/splits/train.txt |
|
val_split: hls_burn_scars/splits/val.txt |
|
test_split: hls_burn_scars/splits/test.txt |
|
img_grep: "*_merged.tif" |
|
label_grep: "*.mask.tif" |
|
means: |
|
- 0.033349706741586264 |
|
- 0.05701185520536176 |
|
- 0.05889748132001316 |
|
- 0.2323245113436119 |
|
- 0.1972854853760658 |
|
- 0.11944914225186566 |
|
stds: |
|
- 0.02269135568823774 |
|
- 0.026807560223070237 |
|
- 0.04004109844362779 |
|
- 0.07791732423672691 |
|
- 0.08708738838140137 |
|
- 0.07241979477437814 |
|
num_classes: 2 |
|
train_transform: |
|
- class_path: albumentations.D4 |
|
- class_path: ToTensorV2 |
|
test_transform: |
|
- class_path: ToTensorV2 |
|
|
|
no_data_replace: 0 |
|
no_label_replace: -1 |
|
|
