SatlasPretrain / config.json
zhangrenchao's picture
Update SatlasPretrain model package
aed6f6f verified
Raw
History Blame Contribute Delete
1.4 kB
{
"model_name": "SatlasPretrain",
"model_type": "satlaspretrain",
"architectures": [
"SatlasPretrain"
],
"framework": "PyTorch",
"domain": "earth-observation",
"task": "remote-sensing-image-understanding",
"implementation": {
"entry_point": "model/satlaspretrain.py",
"scope": "dual-resolution temporal hierarchy with feature pyramid fusion and seven remote-sensing task heads"
},
"architecture": {
"family": "dual-sensor hierarchical encoder with temporal pooling and feature pyramid network",
"highres_time_steps": 4,
"highres_channels": 3,
"highres_image_shape": [
512,
512
],
"lowres_time_steps": 8,
"lowres_channels": 9,
"lowres_image_shape": [
512,
512
],
"embed_dim": 8,
"window_size": 4,
"segmentation_classes": 4,
"property_classes": 3,
"classification_classes": 3,
"task_heads": [
"segmentation",
"regression",
"point",
"polygon",
"polyline",
"property",
"classification"
]
},
"data": {
"datasets": [
"SatlasPretrain",
"NAIP",
"Sentinel-2"
],
"protocol": "satlas_naip4_sentinel2_8_512_npz_v2",
"input_format": "BTCHW",
"train_samples": 1,
"test_samples": 1
},
"configuration_sources": [
"configuration.json",
"conf/config.yaml",
"model/satlaspretrain.py"
]
}