{
    "type": "Feature",
    "stac_version": "1.1.0",
    "stac_extensions": [
        "https://stac-extensions.github.io/mlm/v1.4.0/schema.json"
    ],
    "id": "CloudEnsemble",
    "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 ensemble model for cloud detection in PROBA and SPOTVGT images. It is a combination of five models: 1DPWDeepLabV3, 1DPWSeg, 1DPWUnetPP, Unet, and UnetPP. The model is trained on PROBA-V and SPOTVGT images.",
        "dependencies": [
            "torch",
            "safetensors.torch",
            "semantic-segmentation-models-pytorch"
        ],
        "mlm:framework": "pytorch",
        "mlm:framework_version": "2.1.2+cu121",
        "file:size": 40455416,
        "mlm:memory_size": 1,
        "mlm:accelerator": "cuda",
        "mlm:accelerator_constrained": false,
        "mlm:accelerator_summary": "Unknown",
        "mlm:name": "SegFormerPlusMLP",
        "mlm:architecture": "SegFormer and MLP",
        "mlm:tasks": [
            "cloud detection"
        ],
        "mlm:input": [
            {
                "name": "ProbaVGT and SPOTVGT images",
                "bands": [
                    "Blue[B0]",
                    "Red[B1]",
                    "Near-Infrared[B3]",
                    "SWIR[MIR]"
                ],
                "input": {
                    "shape": [
                        -1,
                        4,
                        128,
                        128
                    ],
                    "dim_order": [
                        "batch",
                        "channel",
                        "height",
                        "width"
                    ],
                    "data_type": "float32"
                },
                "pre_processing_function": null
            }
        ],
        "mlm:output": [
            {
                "name": "cloud mask",
                "tasks": [
                    "cloud detection"
                ],
                "result": {
                    "shape": [
                        -1,
                        1,
                        128,
                        128
                    ],
                    "dim_order": [
                        "batch",
                        "channel",
                        "height",
                        "width"
                    ],
                    "data_type": "uint8"
                },
                "classification:classes": [],
                "post_processing_function": null
            }
        ],
        "mlm:total_parameters": 12894526,
        "mlm:pretrained": true,
        "datetime": null
    },
    "links": [],
    "assets": {
        "1dpwdeeplabv3": {
            "href": "https://huggingface.co/tacofoundation/PROBAandSPOT/resolve/main/ensemble/1dpwdeeplabv3.safetensor",          
            "type": "application/octet-stream; application=safetensor",
            "title": "Pytorch model weights checkpoint",
            "description": "The weights of the model in safetensor format.",
            "mlm:artifact_type": "safetensor.torch.save_file",
            "roles": [
                "mlm:model",
                "mlm:weights",
                "data"
            ]
        },
        "1dpwseg": {
            "href": "https://huggingface.co/tacofoundation/PROBAandSPOT/resolve/main/ensemble/1dpwseg.safetensor",
            "type": "application/octet-stream; application=safetensor",
            "title": "Pytorch model weights checkpoint",
            "description": "The weights of the model in safetensor format.",
            "mlm:artifact_type": "safetensor.torch.save_file",
            "roles": [
                "mlm:model",
                "mlm:weights",
                "data"
            ]
        },
        "1dpwunetpp": {
            "href": "https://huggingface.co/tacofoundation/PROBAandSPOT/resolve/main/ensemble/1dpwunetpp.safetensor",
            "type": "application/octet-stream; application=safetensor",
            "title": "Pytorch model weights checkpoint",
            "description": "The weights of the model in safetensor format.",
            "mlm:artifact_type": "safetensor.torch.save_file",
            "roles": [
                "mlm:model",
                "mlm:weights",
                "data"
            ]
        },
        "unet": {
            "href": "https://huggingface.co/tacofoundation/PROBAandSPOT/resolve/main/ensemble/unet.safetensor",
            "type": "application/octet-stream; application=safetensor",
            "title": "Pytorch model weights checkpoint",
            "description": "The weights of the model in safetensor format.",
            "mlm:artifact_type": "safetensor.torch.save_file",
            "roles": [
                "mlm:model",
                "mlm:weights",
                "data"
            ]
        },
        "unetpp": {
            "href": "https://huggingface.co/tacofoundation/PROBAandSPOT/resolve/main/ensemble/unetpp.safetensor",
            "type": "application/octet-stream; application=safetensor",
            "title": "Pytorch model weights checkpoint",
            "description": "The weights of the model in safetensor format.",
            "mlm:artifact_type": "safetensor.torch.save_file",
            "roles": [
                "mlm:model",
                "mlm:weights",
                "data"
            ]
        },
        "c2r1km": {
            "href": "https://huggingface.co/tacofoundation/PROBAandSPOT/resolve/main/ensemble/c2r1km.safetensor",
            "type": "application/octet-stream; application=safetensor",
            "title": "Pytorch model weights checkpoint",
            "description": "The weights of the model in safetensor format.",
            "mlm:artifact_type": "safetensor.torch.save_file",
            "roles": [
                "mlm:model",
                "mlm:weights",
                "data"
            ]
        },      
        "source_code": {
            "href": "https://huggingface.co/tacofoundation/PROBAandSPOT/resolve/main/ensemble/load.py",
            "type": "text/x-python",
            "title": "Model load script",
            "description": "Python script to load the model.",
            "roles": [
                "mlm:source_code",
                "code"
            ]
        },
        "source_code_extra": {
            "href": "https://huggingface.co/tacofoundation/PROBAandSPOT/resolve/main/ensemble/model.py",
            "type": "text/x-python",
            "title": "Model load script - Extra",
            "description": "Python script to load the model.",
            "roles": [
                "mlm:source_code",
                "code"
            ]
        },
        "source_code_c2r1km": {
            "href": "https://huggingface.co/tacofoundation/PROBAandSPOT/resolve/main/ensemble/c2r1km.py",
            "type": "text/x-python",
            "title": "Model load script - Extra",
            "description": "Python script to load the model.",
            "roles": [
                "mlm:source_code",
                "code"
            ]
        },
        "example_data": {
            "href": "https://huggingface.co/tacofoundation/PROBAandSPOT/resolve/main/ensemble/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": "CloudEnsemble"
}