3D Brain Tumor Segmentation
End-to-end 3D segmentation on BraTS 2019 multi-modal MRI. A 3D U-Net is benchmarked against a Swin-UNETR transformer, then both are served through FastAPI with a Streamlit UI for MRI upload, slice overlays and downloadable masks.
- 0.85 whole-tumor Dice for the 3D U-Net, against 0.74 for Swin-UNETR
- Per-class Dice of 0.77, 0.84 and 0.80 on necrotic core, edema and enhancing tumor
- Class-weighted Dice plus cross-entropy loss for the tumor/background imbalance