All projects
PyTorch
MONAI
Medical AI
Segmentation

3D Lung Tumour Segmentation

·
3D Lung Tumour Segmentation

Overview

3D Lung Tumour Segmentation applies deep learning to medical imaging — specifically, segmenting lung tumours from volumetric CT scans using the Medical Segmentation Decathlon lung dataset.

The model is built with MONAI's 3D U-Net architecture and trained with PyTorch Lightning for structured, reproducible training.

Stack

  • MONAI — medical imaging transforms and 3D U-Net
  • PyTorch Lightning — training loop and checkpointing
  • Hydra — configuration management via conf/config.yaml
  • FastAPI — inference REST API (app/main.py)

Training

python -m venv venv && source venv/bin/activate
pip install -r requirements.txt
python train.py