wolny / pytorch-3dunet

3D U-Net model for volumetric semantic segmentation written in pytorch