| { |
| "architectures": [ |
| "Terratorch" |
| ], |
| "num_classes": 0, |
| "pretrained_cfg": { |
| "seed_everything": 0, |
| "input":{ |
| "target": "pixel_values", |
| "data":{ |
| "pixel_values":{ |
| "type": "torch.Tensor", |
| "shape": [6, 512, 512] |
| }, |
| "location_coords":{ |
| "type":"torch.Tensor", |
| "shape": [1, 2] |
| } |
| } |
| }, |
| "model": { |
| "class_path": "terratorch.tasks.SemanticSegmentationTask", |
| "init_args": { |
| "model_args": { |
| "backbone_pretrained": true, |
| "backbone": "prithvi_eo_v2_300_tl", |
| "decoder": "UperNetDecoder", |
| "decoder_channels": 256, |
| "decoder_scale_modules": true, |
| "num_classes": 2, |
| "rescale": true, |
| "backbone_bands": [ |
| "BLUE", |
| "GREEN", |
| "RED", |
| "NIR_NARROW", |
| "SWIR_1", |
| "SWIR_2" |
| ], |
| "head_dropout": 0.1, |
| "necks": [ |
| { |
| "name": "SelectIndices", |
| "indices": [ |
| 5, |
| 11, |
| 17, |
| 23 |
| ] |
| }, |
| { |
| "name": "ReshapeTokensToImage" |
| } |
| ] |
| }, |
| "model_factory": "EncoderDecoderFactory", |
| "loss": "ce", |
| "ignore_index": -1, |
| "lr": 0.001, |
| "freeze_backbone": false, |
| "freeze_decoder": false, |
| "plot_on_val": 10 |
| } |
| }, |
| "data": { |
| "class_path": "terratorch.datamodules.Sen1Floods11NonGeoDataModule", |
| "init_args": { |
| "data_root": "/dccstor/geofm-finetuning/datasets/sen1floods11", |
| "batch_size": 16, |
| "num_workers": 8, |
| "bands": [ |
| "BLUE", |
| "GREEN", |
| "RED", |
| "NIR_NARROW", |
| "SWIR_1", |
| "SWIR_2" |
| ], |
| "train_transform": [ |
| { |
| "class_path": "albumentations.RandomCrop", |
| "init_args": { |
| "height": 224, |
| "width": 224, |
| "p": 1 |
| } |
| }, |
| { |
| "class_path": "albumentations.HorizontalFlip", |
| "init_args": { |
| "p": 0.5 |
| } |
| }, |
| { |
| "class_path": "albumentations.VerticalFlip", |
| "init_args": { |
| "p": 0.5 |
| } |
| }, |
| { |
| "class_path": "albumentations.pytorch.ToTensorV2", |
| "init_args": { |
| "transpose_mask": false, |
| "p": 1 |
| } |
| } |
| ], |
| "val_transform": [ |
| { |
| "class_path": "albumentations.pytorch.ToTensorV2", |
| "init_args": { |
| "transpose_mask": false, |
| "p": 1 |
| } |
| } |
| ], |
| "test_transform": [ |
| { |
| "class_path": "albumentations.pytorch.ToTensorV2", |
| "init_args": { |
| "transpose_mask": false, |
| "p": 1 |
| } |
| } |
| ], |
| "drop_last": true, |
| "constant_scale": 0.0001, |
| "no_data_replace": 0, |
| "no_label_replace": -1, |
| "use_metadata": false |
| } |
| } |
| } |
| } |
