← all repositories

wolny/pytorch-3dunet

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

2.4k stars Jupyter Notebook Computer VisionDomain Apps
pytorch-3dunet
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.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.