/a-unet

A toolbox that provides hackable building blocks for generic 1D/2D/3D UNets, in PyTorch.

Primary LanguagePythonMIT LicenseMIT

Stargazers