3D U-Net model for volumetric semantic segmentation written in pytorch
Object Classification Training Framework