Skip to content

ayyp1/MRI-Tumor-Segmentation

Repository files navigation

🧠 Brain Tumor Segmentation

This project focuses on brain tumor segmentation using deep learning techniques, specifically leveraging a fine-tuned UNet architecture.

📌 Requirements

All necessary dependencies can be installed using the requirements.txt file:

pip install -r requirements.txt

📂 Dataset

The dataset used for this project can be accessed via Kaggle. It contains MRI scans with corresponding tumor segmentation masks.

🏗 Model

A UNet model was fine-tuned on the above dataset for tumor segmentation. The trained model is available in this repository:

📍 Best Model Checkpoint

🎨 Graphical User Interface (GUI)

To run the GUI, follow these simple steps:

1️⃣ Clone this repository:

git clone https://github.com/your-repo/brain-tumor-segmentation.git
cd brain-tumor-segmentation

2️⃣ Install dependencies:

pip install -r requirements.txt

3️⃣ Run the GUI using Streamlit:

streamlit run main.py

📸 Snapshots

Home Screen Output Screen
Home Output

📚 Training

For training details, refer to the Training Notebook.


If you find this project useful, consider giving it a star!

About

Brain tumor detection app using pytorch for training and streamlit for GUI

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published