| { |
| "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" |
| ] |
| } |
|
|
