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