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