← all repositories

ellisdg/3DUnetCNN

A PyTorch-based 3D U-Net convolutional neural network for volumetric medical image segmentation, used for tasks like brain tumor segmentation.

2.2k stars Python Computer VisionML Frameworks
3DUnetCNN
Velocity · 7d
+0.6
★ / day
Trend
steady
star history

This repository provides a deep learning framework for volumetric medical image segmentation using a 3D U-Net convolutional neural network architecture. It is built with PyTorch and designed to make training and applying deep learning models to medical imaging data accessible and controllable. The project includes tutorials for common medical imaging challenges such as the BraTS brain tumor segmentation dataset, with support for custom configurations and data pipelines.

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