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

Export to BMZ and other changes #108

Merged
merged 14 commits into from
Apr 27, 2024
Merged

Export to BMZ and other changes #108

merged 14 commits into from
Apr 27, 2024

Conversation

jdeschamps
Copy link
Member

Change of behaviour

  • Mean and std are no longer calculated during prediction.
  • Prediction can now be performed without tiling, default behaviour.
  • Prevent extra parameters in TransformModels, which was previously causing wrongly passed parameters to not raise an error.

Configuration

  • ArchitectureModel to simplify models (UNet, VAE, Custom), now export to dict without architecture.

Model I/O

  • HyperParametersCallback to save the configuration with mean and std in the checkpoints.

Others

  • TileInformation object is passed along the tiled prediction, replacement to tuple of arrays.
  • Normalization and denormalization are now compatible with BMZ (using a eps=1e-6 parameter), and are now Albumentations classes.

⚠️ Addded failing tests for depth=3 in UNet and channel > 1. See test_lightning_module.py.

@CatEek CatEek merged commit 6405066 into dl4mia Apr 27, 2024
2 of 15 checks passed
@CatEek CatEek deleted the bmz_io branch April 27, 2024 15:19
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants