Imported from https://github.com/zhengyang-wang/3D-Unet--Tensorflow