Imported from https://github.com/mateuszbuda/brain-segmentation-pytorch