Spaces:
Sleeping
Sleeping
JANGALA SAKETH commited on
Update README.md
Browse files
README.md
CHANGED
|
@@ -16,38 +16,30 @@ You can download sample NIfTI files for two patients to test the model from this
|
|
| 16 |
|
| 17 |
[Sample Data (Google Drive)](https://drive.google.com/drive/folders/19LzKOcoIrWQhwY91e_kn644AcQi4tl8z?usp=sharing)
|
| 18 |
|
| 19 |
-
---
|
| 20 |
-
title: Brain Tumor Segmentation
|
| 21 |
-
emoji: 🧠
|
| 22 |
-
colorFrom: blue
|
| 23 |
-
colorTo: pink
|
| 24 |
-
sdk: streamlit
|
| 25 |
-
sdk_version: 1.48.1
|
| 26 |
-
app_file: app.py
|
| 27 |
-
pinned: false
|
| 28 |
-
license: mit
|
| 29 |
-
---
|
| 30 |
|
| 31 |
# Brain Tumor Segmentation App
|
| 32 |
|
|
|
|
| 33 |
<p align="center">
|
| 34 |
-
|
| 35 |
-
|
| 36 |
</p>
|
| 37 |
|
| 38 |
-
This project is a web application for brain tumor segmentation from 3D/4D NIfTI MRI scans using a 3D U-Net model, built with PyTorch and Streamlit.
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 39 |
|
| 40 |
## Features
|
| 41 |
- Upload four 3D NIfTI brain scans (T1, T1ce, T2, FLAIR)
|
| 42 |
- Automatic preprocessing and patch-based inference
|
| 43 |
- Visualizes the predicted tumor mask overlayed on the MRI
|
| 44 |
|
| 45 |
-
## Quick Start (Hugging Face Spaces)
|
| 46 |
-
|
| 47 |
-
1. **Upload your trained model file** (`unet3d_model.pth`) to the Space's root directory or use a download link in the code.
|
| 48 |
-
2. Click "Run" or "Duplicate Space" to use your own model.
|
| 49 |
-
3. Use the web interface to upload your NIfTI files and view results.
|
| 50 |
-
|
| 51 |
## Local Usage
|
| 52 |
1. Clone this repository:
|
| 53 |
```sh
|
|
|
|
| 16 |
|
| 17 |
[Sample Data (Google Drive)](https://drive.google.com/drive/folders/19LzKOcoIrWQhwY91e_kn644AcQi4tl8z?usp=sharing)
|
| 18 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 19 |
|
| 20 |
# Brain Tumor Segmentation App
|
| 21 |
|
| 22 |
+
|
| 23 |
<p align="center">
|
| 24 |
+
<img src="https://img.shields.io/badge/Streamlit-Online-brightgreen" alt="Streamlit">
|
| 25 |
+
<a href="https://huggingface.co/spaces/saketh-005/brain-tumor-segmentation"><img src="https://img.shields.io/badge/HuggingFace-Live%20Demo-yellow" alt="Hugging Face Spaces"></a>
|
| 26 |
</p>
|
| 27 |
|
| 28 |
+
This project is a web application for brain tumor segmentation from 3D/4D NIfTI MRI scans using a 3D U-Net model, built with PyTorch and Streamlit.
|
| 29 |
+
|
| 30 |
+
## Live Demo
|
| 31 |
+
|
| 32 |
+
👉 **Try the app instantly on Hugging Face Spaces:**
|
| 33 |
+
|
| 34 |
+
[https://huggingface.co/spaces/saketh-005/brain-tumor-segmentation](https://huggingface.co/spaces/saketh-005/brain-tumor-segmentation)
|
| 35 |
+
|
| 36 |
+
No installation required. Just open the link, upload your NIfTI files, and view the results.
|
| 37 |
|
| 38 |
## Features
|
| 39 |
- Upload four 3D NIfTI brain scans (T1, T1ce, T2, FLAIR)
|
| 40 |
- Automatic preprocessing and patch-based inference
|
| 41 |
- Visualizes the predicted tumor mask overlayed on the MRI
|
| 42 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 43 |
## Local Usage
|
| 44 |
1. Clone this repository:
|
| 45 |
```sh
|