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