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

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.