fitzgeraldja

In [this line](https://github.com/fuzailpalnak/building-footprint-segmentation/blame/75f6d6b815efed34f4d7a685dbf616bab18f0f04/examples/PredictionWithAugmentations.ipynb#L219) in the main example notebook, you need to change to ```python transformation.restore_fragment(prediction_binary.swapaxes(1, 2)) ``` so that the restored image correctly aligns with the original input - you...