G-Research / fasttrackml

Experiment tracking server focused on speed and scalability
https://fasttrackml.io/
Apache License 2.0
98 stars 18 forks source link

Aim Images: New Image Logging Endpoint #1231

Closed suprjinx closed 2 months ago

suprjinx commented 3 months ago

New endpoint in MLflow API. It will permit logging an image path (s3 etc) and the necessary metadata (height, width, etc). The exact attributes are as follows:

  1. caption: string
  2. index: int
  3. width: int
  4. height: int
  5. format: string
  6. blob_uri: string (the uri in artifact storage)
  7. run_id: string
  8. step: int
  9. iter: int

This involves DAO, service, and controller layers, with request and response structs. Possibly we will need a roll-up table like latest_metrics to make queries manageable.