GitHub repository: marius-sm/nd_unet
Primary language: Python
Description: Flexible 1D, 2D, and 3D U-Nets in PyTorch