mlm-stac

Wherobots MLM Models

This is a collection of example models implementing the Machine Learning Model Extension to the SpatioTemporal Asset Catalog (STAC) spec. Each metadata json describes a corresponding model asset and the requirements to run that model. These examples, and APIs built on top of them, show the utility of describing data linead and runtime requirements of ML models.

The eventual goal is that most geospatial ML models are represented by MLM metadata, making it easy to run them on STAC datasets and derive more value.

See the MLM spec description if you want to learn more about the MLM description fields: https://github.com/crim-ca/mlm-extension

And also check out the stac-model package for creating and validating MLM metadata: https://github.com/crim-ca/mlm-extension/blob/main/README_STAC_MODEL.md

Each of these models is hosted and deployed in WherobotsAI Raster Inference, a tool for scaling ML models to planet-scale inference tasks.

Big thanks to SATLAS and TorchGeo for distributing open source code and weights for these models.