wolny/pytorch-3dunet
A PyTorch library providing 3D U-Net and residual variants for semantic segmentation of volumetric medical imaging data.

Velocity · 7d
+0.9
★ / day
Trend
→steady
star history
This repository implements 3D U-Net architectures including standard, residual, and squeeze-and-excitation variants for volumetric semantic segmentation. It supports both binary and multi-class segmentation as well as regression tasks like de-noising. The models have been applied to connectomics and medical volume datasets, with configurations available for both 3D and 2D segmentation tasks.