| { |
| "model_name": "Scale-MAE", |
| "model_type": "scale_mae", |
| "architectures": [ |
| "ScaleMAE" |
| ], |
| "framework": "PyTorch", |
| "domain": "earth-observation", |
| "task": "multiscale-geospatial-representation-learning", |
| "implementation": { |
| "entry_point": "model/scalemae.py", |
| "scope": "GSD-aware masked autoencoding with paired low- and high-frequency target reconstruction", |
| "train_script": "scripts/train.py", |
| "inference_script": "scripts/inference.py", |
| "evaluation_script": "scripts/result.py", |
| "synthetic_data_script": "scripts/fake_data.py" |
| }, |
| "architecture": { |
| "family": "scale-aware masked autoencoder", |
| "input_size": 16, |
| "target_size": 32, |
| "patch_size": 4, |
| "input_patches": 16, |
| "in_channels": 3, |
| "embed_dim": 64, |
| "encoder_depth": 2, |
| "encoder_heads": 4, |
| "decoder_dim": 48, |
| "decoder_depth": 1, |
| "decoder_heads": 4, |
| "mask_ratio": 0.75, |
| "reference_gsd": 1.0, |
| "blur_kernel": 5, |
| "band_config": { |
| "low": {"kernel": 5, "target": "low_frequency"}, |
| "high": {"residual": true, "target": "high_frequency"} |
| } |
| }, |
| "data": { |
| "datasets": [ |
| "FMoW-RGB", |
| "RESISC-45", |
| "EuroSAT", |
| "UCMerced", |
| "AID", |
| "MLRSNet" |
| ], |
| "protocol": "synthetic_multiscale_geospatial_rgb", |
| "format": "NPZ", |
| "input_key": "images", |
| "input_shape": ["N", 3, 16, 16], |
| "target_key": "targets", |
| "target_shape": ["N", 3, 32, 32], |
| "gsd_key": "gsd", |
| "gsd_shape": ["N"], |
| "gsd_values": [0.5, 1.0, 2.0, 4.0], |
| "label_key": "labels", |
| "label_shape": ["N"], |
| "num_classes": 4, |
| "default_train_file": "data/train.npz", |
| "default_test_file": "data/test.npz" |
| }, |
| "configuration_sources": [ |
| "conf/config.yaml", |
| "model/scalemae.py", |
| "scripts/fake_data.py", |
| "scripts/train.py", |
| "scripts/inference.py", |
| "scripts/result.py" |
| ] |
| } |
|
|
