README.md

Semantic Segmentation

Final Project for Intro To Deep Learning ECGR 5106

Use segmentation to determine and locate tumors in MRI scans.

Path for Training a Semantic Segmentation Model with Encoder-Decoder

1. Simple CNN Encoder-Decoder
2. Deep CNN Encoder-Decoder
3. ResNet Encoder-Decoder with Batchnorm & Dropout

Example of a Semantic Segmentation

Image Source: cnvrg.io

Classification vs Segmentation

Image Source: Frontiers in Oncology

Resources

• Encoder-Decoder Network: Understanding U-Net
• ConvNeXt Model Code: GitHub PyTorch-Segmentation
• Pre-Processing Data for CNN: Applying CNN on Your Images with PyTorch
• Dataset: Kaggle Brain Tumor Image Dataset