GitHub - cosmic-cortex/pytorch-UNet: 2D and 3D UNet implementation in ...