balbasty / nitorch

Neuroimaging in PyTorch
Other
86 stars 14 forks source link

Segmentation Net #32

Closed balbasty closed 3 years ago

balbasty commented 3 years ago

Basic segmentation UNet. The use of batch normalization improves dramatically its trainability on a dataset with class imbalance.