Imported from https://github.com/pykao/Modified-3D-UNet-Pytorch