3D U-Net model for volumetric semantic segmentation written in pytorch - View it on GitHub