rybavery's picture
example MLM v1.2.0 model metadata and artifacts
f4fec64
{
"type": "Feature",
"stac_version": "1.0.0",
"id": "item_solar_satlas_sentinel2",
"properties": {
"start_datetime": "1900-01-01T00:00:00Z",
"end_datetime": "9999-01-01T00:00:00Z",
"description": "Sourced from satlas source code released by Allen AI under Apache 2.0",
"mlm:framework": "pytorch",
"mlm:framework_version": "2.3.0+cu121",
"file:size": 333000000,
"mlm:memory_size": 1,
"mlm:batch_size_suggestion": 10,
"mlm:accelerator": "cuda",
"mlm:accelerator_constrained": true,
"mlm:accelerator_summary": "It is necessary to use GPU since it was compiled for NVIDIA Ampere and newer architectures with AOTInductor and the computational demands of the model.",
"mlm:name": "Satlas Solar Farm Segmentation",
"mlm:architecture": "Swin Transformer V2 with U-Net head",
"mlm:tasks": [
"semantic-segmentation",
"segmentation"
],
"mlm:input": [
{
"name": "9 Band Sentinel-2 4 Time Step Series Batch",
"bands": [
"B02",
"B03",
"B04",
"B05",
"B06",
"B07",
"B08",
"B11",
"B12",
"B02",
"B03",
"B04",
"B05",
"B06",
"B07",
"B08",
"B11",
"B12",
"B02",
"B03",
"B04",
"B05",
"B06",
"B07",
"B08",
"B11",
"B12",
"B02",
"B03",
"B04",
"B05",
"B06",
"B07",
"B08",
"B11",
"B12"
],
"input": {
"shape": [
-1,
36,
1024,
1024
],
"dim_order": [
"batch",
"channel",
"height",
"width"
],
"data_type": "float32"
},
"norm_by_channel": true,
"norm_type": "min-max",
"resize_type": "crop",
"statistics": [
{
"minimum": 0,
"maximum": 255
},
{
"minimum": 0,
"maximum": 255
},
{
"minimum": 0,
"maximum": 255
},
{
"minimum": 0,
"maximum": 255
},
{
"minimum": 0,
"maximum": 255
},
{
"minimum": 0,
"maximum": 255
},
{
"minimum": 0,
"maximum": 255
},
{
"minimum": 0,
"maximum": 255
},
{
"minimum": 0,
"maximum": 255
},
{
"minimum": 0,
"maximum": 255
},
{
"minimum": 0,
"maximum": 255
},
{
"minimum": 0,
"maximum": 255
},
{
"minimum": 0,
"maximum": 255
},
{
"minimum": 0,
"maximum": 255
},
{
"minimum": 0,
"maximum": 255
},
{
"minimum": 0,
"maximum": 255
},
{
"minimum": 0,
"maximum": 255
},
{
"minimum": 0,
"maximum": 255
},
{
"minimum": 0,
"maximum": 255
},
{
"minimum": 0,
"maximum": 255
},
{
"minimum": 0,
"maximum": 255
},
{
"minimum": 0,
"maximum": 255
},
{
"minimum": 0,
"maximum": 255
},
{
"minimum": 0,
"maximum": 255
},
{
"minimum": 0,
"maximum": 255
},
{
"minimum": 0,
"maximum": 255
},
{
"minimum": 0,
"maximum": 255
},
{
"minimum": 0,
"maximum": 255
},
{
"minimum": 0,
"maximum": 255
},
{
"minimum": 0,
"maximum": 255
},
{
"minimum": 0,
"maximum": 255
},
{
"minimum": 0,
"maximum": 255
},
{
"minimum": 0,
"maximum": 255
},
{
"minimum": 0,
"maximum": 255
},
{
"minimum": 0,
"maximum": 255
},
{
"minimum": 0,
"maximum": 255
}
],
"pre_processing_function": {
"format": "documentation-link",
"expression": "https://github.com/allenai/satlas/blob/main/CustomInference.md#sentinel-2-inference-example"
}
}
],
"mlm:output": [
{
"name": "confidence array",
"tasks": [
"semantic-segmentation"
],
"result": {
"shape": [
-1,
1,
1024,
1024
],
"dim_order": [
"batch",
"height",
"width"
],
"data_type": "float32"
},
"classification:classes": [
{
"value": 1,
"name": "Solar Farm",
"description": "Solar Farm"
}
],
"post_processing_function": null
}
],
"mlm:total_parameters": 89748193,
"mlm:pretrained": true,
"mlm:pretrained_source": "Sentinel-2 imagery and SATLAS labels",
"datetime": null
},
"geometry": {
"type": "Polygon",
"coordinates": [
[
[
-7.882190080512502,
37.13739173208318
],
[
-7.882190080512502,
58.21798141355221
],
[
27.911651652899923,
58.21798141355221
],
[
27.911651652899923,
37.13739173208318
],
[
-7.882190080512502,
37.13739173208318
]
]
]
},
"links": [
{
"rel": "derived_from",
"href": "https://earth-search.aws.element84.com/v1/collections/sentinel-2-l1c",
"type": "application/json"
},
{
"rel": "self",
"href": "s3://wherobots-modelhub-prod/professional/semantic-segmentation/solar-satlas-sentinel2/model-metadata.json/item_solar_satlas_sentinel2.json",
"type": "application/json"
}
],
"assets": {
"model": {
"href": "s3://wherobots-modelhub-prod/professional/semantic-segmentation/solar-satlas-sentinel2/inductor/gpu/aot_inductor_gpu_tensor_cores.zip",
"type": "application/zip; application=pytorch",
"title": "AOTInductor model exported from private, edited, hard fork of Satlas github repo.",
"description": "A Swin Transformer backbone with a U-net head trained on the 9-band Sentinel-2 Top of Atmosphere product.",
"mlm_artifact_type": "torch.jit.script",
"file:size": 333000000,
"roles": [
"mlm:model",
"data"
]
},
"source_code": {
"href": "https://github.com/wherobots/modelhub/blob/main/model-forge/satlas/solar/export.py",
"type": "text/x-python",
"title": "Model implementation.",
"description": "Source code to export the model.",
"roles": [
"mlm:model",
"code"
]
}
},
"bbox": [
-7.882190080512502,
37.13739173208318,
27.911651652899923,
58.21798141355221
],
"stac_extensions": [
"https://stac-extensions.github.io/file/v2.1.0/schema.json",
"https://crim-ca.github.io/mlm-extension/v1.2.0/schema.json"
]
}