The example code for u-net brain MRI may not include normalization pre-step about hub HOT 2 OPEN

@mateuszbuda would you mind taking a quick look? Thanks!

from hub.

@Anonymousjoy Thanks for trying my model.
You're right that the example code does not produce correct segmentations due to a simplification in the normalization step. The reason for this is that the model was trained to perform segmentation of 3D MRI volumes by concatenating predictions for 2D slices. The example code only shows how to forward one 2D slice.
The normalization step requires the entire volume which I didn't want to upload on GitHub for the sake of data size.
Here is a link to normalization function that works on volumes: https://github.com/mateuszbuda/brain-segmentation-pytorch/blob/8ef2e2d423b67b53ec8113fc71a9b968bb0f66e7/utils.py#L77

from hub.