Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Add prediction writer callback #323

Open
7 tasks
melisande-c opened this issue Dec 10, 2024 · 0 comments
Open
7 tasks

Add prediction writer callback #323

melisande-c opened this issue Dec 10, 2024 · 0 comments
Assignees
Labels
feature New feature or request

Comments

@melisande-c
Copy link
Member

The idea of creating a lightning prediction writer callback is to unify writing to TIFF and writing to ZARR.

The strategy for writing a to the different files can be decoupled from the prediction writer itself by creating write strategy interface, with concrete classes that can be an attribute of the prediction writer.

TODO:

  • PredictionWriterCallback
  • write strategies
    • WriteImage
    • WriteTiles
    • WriteZarrTiles
  • File metadata extractors (mostly to extract the number of samples (i.e. tiff pages) per file)
  • Reimplement CAREamist.predict_to_disk function
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
feature New feature or request
Projects
None yet
Development

No branches or pull requests

1 participant