OneScience-Group/SatMAE

1{2  "model_name": "SatMAE",3  "model_type": "satmae",4  "architectures": [5    "SatMAE"6  ],7  "framework": "PyTorch",8  "domain": "earth-observation",9  "task": "remote-sensing-representation-learning",10  "implementation": {11    "entry_point": "model/satmae.py",12    "scope": "masked autoencoding for temporal or grouped multispectral satellite imagery",13    "train_script": "scripts/train.py",14    "inference_script": "scripts/inference.py",15    "evaluation_script": "scripts/result.py",16    "synthetic_data_script": "scripts/fake_data.py"17  },18  "architecture": {19    "family": "temporal and grouped multispectral masked autoencoder",20    "runtime_profile": "smoke",21    "mode": "temporal",22    "image_size": 32,23    "patch_size": 4,24    "in_channels": 3,25    "frames": 3,26    "embed_dim": 64,27    "encoder_depth": 2,28    "encoder_heads": 4,29    "decoder_dim": 32,30    "decoder_depth": 1,31    "decoder_heads": 4,32    "mask_ratio": 0.75,33    "spectral_groups": [[0, 1], [2]],34    "norm_pix_loss": false,35    "same_mask": false,36    "spatial_mask": false37  },38  "data": {39    "datasets": [40      "fMoW RGB",41      "fMoW-Sentinel",42      "NAIP",43      "EuroSAT",44      "BigEarthNet",45      "SpaceNet v1"46    ],47    "protocol": "fmow_rgb_temporal",48    "format": "NPZ",49    "input_key": "images",50    "input_shape": ["N", 3, 3, 32, 32],51    "timestamp_key": "timestamps",52    "timestamp_shape": ["N", 3, 3],53    "timestamp_fields": ["year_offset_2002", "month_zero_based", "hour"],54    "label_key": "labels",55    "label_shape": ["N"],56    "num_classes": 10,57    "default_train_file": "data/train.npz",58    "default_test_file": "data/test.npz",59    "required_metadata": ["source_protocol", "data_source"]60  },61  "configuration_sources": [62    "conf/config.yaml",63    "model/satmae.py",64    "scripts/fake_data.py",65    "scripts/train.py",66    "scripts/inference.py",67    "scripts/result.py"68  ]69}70