mlm.json

{
    "type": "Feature",
    "stac_version": "1.1.0",
    "stac_extensions": [
        "https://stac-extensions.github.io/mlm/v1.4.0/schema.json"
    ],
    "id": "UNetMobV2_V2 model",
    "geometry": {
        "type": "Polygon",
        "coordinates": [
            [
                [
                    -180.0,
                    -90.0
                ],
                [
                    -180.0,
                    90.0
                ],
                [
                    180.0,
                    90.0
                ],
                [
                    180.0,
                    -90.0
                ],
                [
                    -180.0,
                    -90.0
                ]
            ]
        ]
    },
    "bbox": [
        -180,
        -90,
        180,
        90
    ],
    "properties": {
        "start_datetime": "1900-01-01T00:00:00Z",
        "end_datetime": "9999-01-01T00:00:00Z",
        "description": "A UNet model trained on Sentinel-2 imagery for cloud segmentation.",
        "forward_backward_pass": {
            "32": 3.229184,
            "64": 12.916736,
            "128": 51.666944,
            "256": 206.667776,
            "512": 826.671104,
            "1024": 3306.684416,
            "2048": 13226.737664
        },
        "dependencies": [
            "torch",
            "segmentation-models-pytorch",
            "safetensors.torch"
        ],
        "mlm:framework": "pytorch",
        "mlm:framework_version": "2.1.2+cu121",
        "file:size": 26529040,
        "mlm:memory_size": 1,
        "mlm:accelerator": "cuda",
        "mlm:accelerator_constrained": false,
        "mlm:accelerator_summary": "Unknown",
        "mlm:name": "UNetMobV2_V1",
        "mlm:architecture": "UNetMobV2",
        "mlm:tasks": [
            "semantic-segmentation"
        ],
        "mlm:input": [
            {
                "name": "13 Band Sentinel-2 Batch",
                "bands": [
                    "B01",
                    "B02",
                    "B03",
                    "B04",
                    "B05",
                    "B06",
                    "B07",
                    "B08",
                    "B8A",
                    "B09",
                    "B10",
                    "B11",
                    "B12"
                ],
                "input": {
                    "shape": [
                        -1,
                        13,
                        512,
                        512
                    ],
                    "dim_order": [
                        "batch",
                        "channel",
                        "height",
                        "width"
                    ],
                    "data_type": "float32"
                },
                "pre_processing_function": null
            }
        ],
        "mlm:output": [
            {
                "name": "semantic-segmentation",
                "tasks": [
                    "semantic-segmentation"
                ],
                "result": {
                    "shape": [
                        -1,
                        4,
                        512,
                        512
                    ],
                    "dim_order": [
                        "batch",
                        "channel",
                        "height",
                        "width"
                    ],
                    "data_type": "float32"
                },
                "classification:classes": [
                    {
                        "value": 0,
                        "name": "Clear",
                        "description": "Clear"
                    },
                    {
                        "value": 1,
                        "name": "Thick Clouds",
                        "description": "Thick Clouds"
                    },
                    {
                        "value": 2,
                        "name": "Thin Clouds",
                        "description": "Thin Clouds"
                    },
                    {
                        "value": 3,
                        "name": "Cloud Shadows",
                        "description": "Cloud Shadows"
                    }
                ],
                "post_processing_function": null
            }
        ],
        "mlm:total_parameters": 6632260,
        "mlm:pretrained": true,
        "datetime": null
    },
    "links": [],
    "assets": {
        "trainable": {
            "href": "https://huggingface.co/tacofoundation/GANFilling/resolve/main/model.safetensor",
            "type": "application/octet-stream; application=safetensor",
            "title": "Pytorch weights checkpoint",
            "description": "A UNet model trained on Sentinel-2 imagery for cloud segmentation.The model was trained using the CloudSEN12 dataset.",
            "mlm:artifact_type": "safetensor.torch.save_file",
            "roles": [
                "mlm:model",
                "mlm:weights",
                "data"
            ]
        },
        "source_code": {
            "href": "https://huggingface.co/tacofoundation/GANFilling/resolve/main/load.py",
            "type": "text/x-python",
            "title": "Model load script",
            "description": "Source code to run the model.",
            "roles": [
                "mlm:source_code",
                "code"
            ]
        },
        "source_code_model": {
            "href": "https://huggingface.co/tacofoundation/GANFilling/resolve/main/model.py",
            "type": "text/x-python",
            "title": "Model load script",
            "description": "Source code to run the model.",
            "roles": [
                "mlm:source_code",
                "code"
            ]
        },        
        "example_data": {
            "href": "https://huggingface.co/tacofoundation/GANFilling/resolve/main/example_data.safetensor",
            "type": "application/octet-stream; application=safetensors",
            "title": "Example Sentinel-2 image",
            "description": "Example Sentinel-2 image for model inference.",
            "roles": [
                "mlm:example_data",
                "data"
            ]
        }
    },
    "collection": "GANFilling"
}