Updates for the Unet model
Remember to rebase with main!
TODO
- Instead of always resizing, check and transform the image to the closest possible shape. In the case of transformation, warn the user.
- Adapt the notebook to model creation before the transformation (transformation needs model as input)
- Do not set level to info inside the function code. It can be placed in the notebook, before the training starts.