allenai / rslearn

A tool for developing remote sensing datasets and models.
Apache License 2.0
5 stars 0 forks source link

Support HTTP API for inference and building Docker container with it #34

Open favyen2 opened 1 month ago

favyen2 commented 1 month ago

There should be an entrypoint to serve an HTTP API that can accept two kinds of requests:

  1. Request specifies a spatiotemporal window. The server should ingest images corresponding to that window, and then run the model on the ingested images.
  2. Request provides the images directly. Then the server should just apply the model on those images.

It should support gridded inference option which should have an abstraction to specify how to merge the results across the grid cells.

The entrypoint should probably take a dataset configuration file (JSON) along with model config file (YAML) as arguments. There may be more configuration too.

Then a Dockerfile should be included making it easy to build Docker container around the entrypoint.

favyen2 commented 1 month ago

How will this work for Satlas models (which may need additional post-processing of the outputs like Viterbi smoothing)?