Spaces:
Runtime error
Runtime error
Upload folder using huggingface_hub
Browse files- .gitignore +40 -0
- LICENSE +21 -0
- README.md +181 -10
- app.py +401 -0
- requirements.txt +30 -0
.gitignore
ADDED
|
@@ -0,0 +1,40 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
# Python
|
| 2 |
+
__pycache__/
|
| 3 |
+
*.py[cod]
|
| 4 |
+
*$py.class
|
| 5 |
+
*.so
|
| 6 |
+
.Python
|
| 7 |
+
|
| 8 |
+
# Virtual Environment
|
| 9 |
+
env/
|
| 10 |
+
venv/
|
| 11 |
+
ENV/
|
| 12 |
+
|
| 13 |
+
# Jupyter
|
| 14 |
+
.ipynb_checkpoints/
|
| 15 |
+
|
| 16 |
+
# Data
|
| 17 |
+
data/raw/
|
| 18 |
+
*.zip
|
| 19 |
+
*.tar.gz
|
| 20 |
+
|
| 21 |
+
# Models (large files)
|
| 22 |
+
*.pth
|
| 23 |
+
*.onnx
|
| 24 |
+
|
| 25 |
+
# Logs
|
| 26 |
+
logs/
|
| 27 |
+
*.log
|
| 28 |
+
wandb/
|
| 29 |
+
|
| 30 |
+
# IDE
|
| 31 |
+
.vscode/
|
| 32 |
+
.idea/
|
| 33 |
+
*.swp
|
| 34 |
+
|
| 35 |
+
# OS
|
| 36 |
+
.DS_Store
|
| 37 |
+
Thumbs.db
|
| 38 |
+
|
| 39 |
+
# Gradio
|
| 40 |
+
flagged/
|
LICENSE
ADDED
|
@@ -0,0 +1,21 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
MIT License
|
| 2 |
+
|
| 3 |
+
Copyright (c) 2025 Abdullah Rubab
|
| 4 |
+
|
| 5 |
+
Permission is hereby granted, free of charge, to any person obtaining a copy
|
| 6 |
+
of this software and associated documentation files (the "Software"), to deal
|
| 7 |
+
in the Software without restriction, including without limitation the rights
|
| 8 |
+
to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
| 9 |
+
copies of the Software, and to permit persons to whom the Software is
|
| 10 |
+
furnished to do so, subject to the following conditions:
|
| 11 |
+
|
| 12 |
+
The above copyright notice and this permission notice shall be included in all
|
| 13 |
+
copies or substantial portions of the Software.
|
| 14 |
+
|
| 15 |
+
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
| 16 |
+
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
| 17 |
+
FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
| 18 |
+
AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
| 19 |
+
LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
| 20 |
+
OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
|
| 21 |
+
SOFTWARE.
|
README.md
CHANGED
|
@@ -1,12 +1,183 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
---
|
| 2 |
-
title: Brain Tumor Classifier
|
| 3 |
-
emoji: 👁
|
| 4 |
-
colorFrom: blue
|
| 5 |
-
colorTo: red
|
| 6 |
-
sdk: gradio
|
| 7 |
-
sdk_version: 5.49.0
|
| 8 |
-
app_file: app.py
|
| 9 |
-
pinned: false
|
| 10 |
-
---
|
| 11 |
|
| 12 |
-
|
|
|
|
|
|
|
|
|
| 1 |
+
# 🧠 Brain Tumor MRI Classification
|
| 2 |
+
|
| 3 |
+
[](https://www.python.org/)
|
| 4 |
+
[](https://pytorch.org/)
|
| 5 |
+
[](https://gradio.app/)
|
| 6 |
+
[](LICENSE)
|
| 7 |
+
|
| 8 |
+
> Advanced Deep Learning System for Automated Brain Tumor Detection and Classification
|
| 9 |
+
|
| 10 |
+
## 🎯 Overview
|
| 11 |
+
|
| 12 |
+
A state-of-the-art deep learning system for classifying brain tumors from MRI scans using ensemble learning and explainable AI. The system achieves **96%+ accuracy** through an ensemble of three powerful models: ConvNeXt, Vision Transformer, and EfficientNetV2.
|
| 13 |
+
|
| 14 |
+
## ✨ Key Features
|
| 15 |
+
|
| 16 |
+
- 🎯 **High Accuracy**: 96%+ ensemble accuracy, 99%+ individual model F1 scores
|
| 17 |
+
- 🔍 **Explainable AI**: Grad-CAM++ visualization for all models
|
| 18 |
+
- 🌐 **Multi-language**: English, Bengali (বাংলা), Spanish (Español)
|
| 19 |
+
- ⚡ **Real-time**: < 1 second inference time
|
| 20 |
+
- 🎨 **Professional UI**: Dark theme Gradio interface
|
| 21 |
+
- 🔄 **Test-Time Augmentation**: Optional accuracy boost
|
| 22 |
+
- 🚀 **Live Demo**: Deployed on HuggingFace Spaces
|
| 23 |
+
|
| 24 |
+
## 🏥 Tumor Types Detected
|
| 25 |
+
|
| 26 |
+
| Type | Description |
|
| 27 |
+
|------|-------------|
|
| 28 |
+
| **Glioma** | Tumor originating from glial cells |
|
| 29 |
+
| **Meningioma** | Tumor of the meninges |
|
| 30 |
+
| **Pituitary** | Tumor in the pituitary gland |
|
| 31 |
+
| **No Tumor** | Healthy brain scan |
|
| 32 |
+
|
| 33 |
+
## 🚀 Try It Live
|
| 34 |
+
|
| 35 |
+
🔗 **[Live Demo on HuggingFace Spaces](#)** *(Link will be added after deployment)*
|
| 36 |
+
|
| 37 |
+
## 🤖 Model Architecture
|
| 38 |
+
|
| 39 |
+
### Ensemble of Three State-of-the-Art Models
|
| 40 |
+
|
| 41 |
+
| Model | Parameters | Val F1 Score | Description |
|
| 42 |
+
|-------|-----------|-------------|-------------|
|
| 43 |
+
| **ConvNeXt-Base** | 87.6M | 99.9% | Modern CNN architecture |
|
| 44 |
+
| **ViT-B/16** | 85.8M | 99%+ | Vision Transformer |
|
| 45 |
+
| **EfficientNetV2-M** | 52.9M | 99%+ | Efficient CNN |
|
| 46 |
+
| **Ensemble** | - | **96%+** | Weighted average (0.4, 0.35, 0.25) |
|
| 47 |
+
|
| 48 |
+
## 📊 Performance Metrics
|
| 49 |
+
|
| 50 |
+
- ✅ **Ensemble Accuracy**: 96%+
|
| 51 |
+
- ✅ **Individual Models**: 99%+ F1 scores
|
| 52 |
+
- ✅ **Inference Time**: < 1 second per image
|
| 53 |
+
- ✅ **Real-time Capable**: Yes
|
| 54 |
+
- ✅ **TTA Boost**: ±2% accuracy improvement
|
| 55 |
+
|
| 56 |
+
## 🔍 Explainable AI
|
| 57 |
+
|
| 58 |
+
The system provides visual explanations through:
|
| 59 |
+
|
| 60 |
+
- **Grad-CAM++**: Heatmaps showing decision regions for CNN models
|
| 61 |
+
- **Attention Maps**: Visualization for Vision Transformer
|
| 62 |
+
- **Confidence Distribution**: Probability breakdown for all classes
|
| 63 |
+
|
| 64 |
+
## 🛠️ Technology Stack
|
| 65 |
+
|
| 66 |
+
**Deep Learning**
|
| 67 |
+
- PyTorch 2.5
|
| 68 |
+
- timm (PyTorch Image Models)
|
| 69 |
+
- Transfer learning from ImageNet-21k
|
| 70 |
+
|
| 71 |
+
**Computer Vision**
|
| 72 |
+
- OpenCV
|
| 73 |
+
- Albumentations (data augmentation)
|
| 74 |
+
- Pillow
|
| 75 |
+
|
| 76 |
+
**UI & Deployment**
|
| 77 |
+
- Gradio 5.0 (professional dark theme)
|
| 78 |
+
- HuggingFace Spaces
|
| 79 |
+
|
| 80 |
+
**Explainability**
|
| 81 |
+
- Grad-CAM++
|
| 82 |
+
- Attention visualization
|
| 83 |
+
|
| 84 |
+
## 📁 Project Structure
|
| 85 |
+
|
| 86 |
+
brain-tumor-mri-classification/
|
| 87 |
+
├── notebooks/ # 6 Kaggle notebooks (complete pipeline)
|
| 88 |
+
├── models/best/ # Trained model checkpoints
|
| 89 |
+
├── app.py # HuggingFace Spaces application
|
| 90 |
+
├── requirements.txt # Python dependencies
|
| 91 |
+
├── README.md # This file
|
| 92 |
+
└── LICENSE # MIT License
|
| 93 |
+
|
| 94 |
+
## 🚀 Quick Start
|
| 95 |
+
|
| 96 |
+
Clone repository
|
| 97 |
+
git clone https://github.com/ABRUBAB/brain-tumor-mri-classification.git
|
| 98 |
+
cd brain-tumor-mri-classification
|
| 99 |
+
|
| 100 |
+
Install dependencies
|
| 101 |
+
pip install -r requirements.txt
|
| 102 |
+
|
| 103 |
+
Run application
|
| 104 |
+
python app.py
|
| 105 |
+
|
| 106 |
+
|
| 107 |
+
## 📝 Training Details
|
| 108 |
+
|
| 109 |
+
**Data Preprocessing**
|
| 110 |
+
- Image size: 224×224
|
| 111 |
+
- Normalization: ImageNet statistics
|
| 112 |
+
- Augmentation: Horizontal/vertical flip, rotation, brightness, contrast, gaussian noise
|
| 113 |
+
|
| 114 |
+
**Training Configuration**
|
| 115 |
+
- Optimizer: AdamW (lr=1e-4, weight_decay=1e-2)
|
| 116 |
+
- Scheduler: CosineAnnealingWarmRestarts
|
| 117 |
+
- Loss: CrossEntropyLoss
|
| 118 |
+
- Early stopping with patience=5
|
| 119 |
+
- Mixed precision training
|
| 120 |
+
|
| 121 |
+
## 📚 Notebooks
|
| 122 |
+
|
| 123 |
+
This project includes 6 comprehensive Kaggle notebooks:
|
| 124 |
+
|
| 125 |
+
1. **Setup & Data Collection** - Environment setup, dataset download
|
| 126 |
+
2. **EDA & Preprocessing** - Exploratory analysis, augmentation pipeline
|
| 127 |
+
3. **Model Training** - Train ConvNeXt, ViT, EfficientNetV2
|
| 128 |
+
4. **Ensemble & Explainability** - Create ensemble, implement Grad-CAM++, build UI
|
| 129 |
+
5. **GitHub Integration** - Create deployment files, documentation
|
| 130 |
+
6. **Final Testing** - Test live deployment, completion verification
|
| 131 |
+
|
| 132 |
+
## 🌐 Multi-language Support
|
| 133 |
+
|
| 134 |
+
The system provides medical reports in:
|
| 135 |
+
- 🇺🇸 **English** - Comprehensive medical interpretation
|
| 136 |
+
- 🇧🇩 **বাংলা (Bengali)** - Full Bengali translation
|
| 137 |
+
- 🇪🇸 **Español (Spanish)** - Complete Spanish translation
|
| 138 |
+
|
| 139 |
+
## ⚠️ Medical Disclaimer
|
| 140 |
+
|
| 141 |
+
**IMPORTANT**: This AI system is developed for **educational and research purposes only**. It should **NOT** be used as a substitute for professional medical diagnosis.
|
| 142 |
+
|
| 143 |
+
- Always seek advice from qualified healthcare providers
|
| 144 |
+
- Never disregard professional medical advice based on AI predictions
|
| 145 |
+
- This system is not FDA/EMA approved
|
| 146 |
+
- Clinical decisions should only be made by certified medical practitioners
|
| 147 |
+
|
| 148 |
+
## 👨💻 Author
|
| 149 |
+
|
| 150 |
+
**Abdullah Rubab**
|
| 151 |
+
|
| 152 |
+
- 🏫 Daffodil International University, Bangladesh
|
| 153 |
+
- 📧 Email: rubab2712@gmail.com
|
| 154 |
+
- 🐙 GitHub: [@ABRUBAB](https://github.com/ABRUBAB)
|
| 155 |
+
|
| 156 |
+
## 📄 License
|
| 157 |
+
|
| 158 |
+
This project is licensed under the MIT License - see the [LICENSE](LICENSE) file for details.
|
| 159 |
+
|
| 160 |
+
## 🙏 Acknowledgments
|
| 161 |
+
|
| 162 |
+
- **Dataset**: Brain Tumor MRI Dataset (Kaggle)
|
| 163 |
+
- **Pretrained Models**: timm library (HuggingFace)
|
| 164 |
+
- **Framework**: PyTorch
|
| 165 |
+
- **UI Framework**: Gradio
|
| 166 |
+
|
| 167 |
+
## 📚 Citation
|
| 168 |
+
|
| 169 |
+
If you use this project in your research, please cite:
|
| 170 |
+
|
| 171 |
+
@software{rubab2025braintumor,
|
| 172 |
+
author = {Rubab, Abdullah},
|
| 173 |
+
title = {Brain Tumor MRI Classification using Deep Learning},
|
| 174 |
+
year = {2025},
|
| 175 |
+
publisher = {GitHub},
|
| 176 |
+
url = {https://github.com/ABRUBAB/brain-tumor-mri-classification}
|
| 177 |
+
}
|
| 178 |
+
|
| 179 |
---
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 180 |
|
| 181 |
+
**⭐ If you found this project helpful, please consider giving it a star!**
|
| 182 |
+
|
| 183 |
+
© 2025 Abdullah Rubab. All rights reserved.
|
app.py
ADDED
|
@@ -0,0 +1,401 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
"""
|
| 2 |
+
Brain Tumor MRI Classification - HuggingFace Spaces Application
|
| 3 |
+
Author: Abdullah Rubab
|
| 4 |
+
Email: rubab2712@gmail.com
|
| 5 |
+
GitHub: @ABRUBAB
|
| 6 |
+
University: Daffodil International University
|
| 7 |
+
"""
|
| 8 |
+
|
| 9 |
+
import gradio as gr
|
| 10 |
+
import torch
|
| 11 |
+
import torch.nn.functional as F
|
| 12 |
+
import numpy as np
|
| 13 |
+
import cv2
|
| 14 |
+
import timm
|
| 15 |
+
from pathlib import Path
|
| 16 |
+
import albumentations as A
|
| 17 |
+
from albumentations.pytorch import ToTensorV2
|
| 18 |
+
from PIL import Image as PILImage
|
| 19 |
+
import matplotlib.pyplot as plt
|
| 20 |
+
|
| 21 |
+
# ===== CONFIGURATION =====
|
| 22 |
+
DEVICE = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
|
| 23 |
+
CLASSES = ['glioma', 'meningioma', 'notumor', 'pituitary']
|
| 24 |
+
|
| 25 |
+
print(f"🚀 Starting Brain Tumor Classifier on {DEVICE}")
|
| 26 |
+
|
| 27 |
+
# ===== IMAGE TRANSFORMS =====
|
| 28 |
+
test_transform = A.Compose([
|
| 29 |
+
A.Resize(224, 224),
|
| 30 |
+
A.Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225]),
|
| 31 |
+
ToTensorV2()
|
| 32 |
+
])
|
| 33 |
+
|
| 34 |
+
# ===== MODEL LOADING =====
|
| 35 |
+
def load_models():
|
| 36 |
+
"""Load trained models from checkpoints"""
|
| 37 |
+
try:
|
| 38 |
+
print("Loading models...")
|
| 39 |
+
|
| 40 |
+
# Note: Upload your .pth files to HuggingFace Space
|
| 41 |
+
# For demo mode, models will be None
|
| 42 |
+
|
| 43 |
+
convnext_model = None
|
| 44 |
+
vit_model = None
|
| 45 |
+
efficientnet_model = None
|
| 46 |
+
ensemble_model = None
|
| 47 |
+
|
| 48 |
+
# TODO: Uncomment and customize when you upload model files
|
| 49 |
+
# convnext_model = timm.create_model('convnext_base', pretrained=False, num_classes=4)
|
| 50 |
+
# convnext_model.load_state_dict(torch.load('models/best/best_convnext_model.pth', map_location=DEVICE))
|
| 51 |
+
# convnext_model.to(DEVICE).eval()
|
| 52 |
+
|
| 53 |
+
print("✅ Models ready (demo mode)")
|
| 54 |
+
return convnext_model, vit_model, efficientnet_model, ensemble_model
|
| 55 |
+
|
| 56 |
+
except Exception as e:
|
| 57 |
+
print(f"⚠️ Model loading: {e}")
|
| 58 |
+
return None, None, None, None
|
| 59 |
+
|
| 60 |
+
convnext_model, vit_model, efficientnet_model, ensemble_model = load_models()
|
| 61 |
+
|
| 62 |
+
# ===== PREDICTION FUNCTION =====
|
| 63 |
+
def predict_brain_tumor(image, model_choice, use_tta, show_explanations, language):
|
| 64 |
+
"""Main prediction function"""
|
| 65 |
+
|
| 66 |
+
if image is None:
|
| 67 |
+
error_msgs = {
|
| 68 |
+
'en': "❌ Please upload an MRI scan image",
|
| 69 |
+
'bn': "❌ অনুগ্রহ করে একটি MRI স্ক্যান ছবি আপলোড করুন",
|
| 70 |
+
'es': "❌ Por favor, sube una imagen de resonancia magnética"
|
| 71 |
+
}
|
| 72 |
+
return error_msgs.get(language, error_msgs['en']), None, None, None
|
| 73 |
+
|
| 74 |
+
try:
|
| 75 |
+
# Preprocess image
|
| 76 |
+
if isinstance(image, str):
|
| 77 |
+
image_pil = PILImage.open(image).convert('RGB')
|
| 78 |
+
else:
|
| 79 |
+
image_pil = image.convert('RGB')
|
| 80 |
+
|
| 81 |
+
img = cv2.cvtColor(np.array(image_pil), cv2.COLOR_RGB2BGR)
|
| 82 |
+
img_resized = cv2.resize(img, (224, 224))
|
| 83 |
+
img_rgb = cv2.cvtColor(img_resized, cv2.COLOR_BGR2RGB)
|
| 84 |
+
|
| 85 |
+
# Demo mode report (replace with actual prediction when models are loaded)
|
| 86 |
+
if convnext_model is None:
|
| 87 |
+
report = generate_demo_report(model_choice, language)
|
| 88 |
+
|
| 89 |
+
# Create demo chart
|
| 90 |
+
fig, ax = plt.subplots(figsize=(10, 4.5), facecolor='#1E1E1E')
|
| 91 |
+
demo_probs = [0.25, 0.25, 0.25, 0.25] # Demo values
|
| 92 |
+
colors = ['#00D9FF', '#4A5568', '#4A5568', '#4A5568']
|
| 93 |
+
class_labels = ['Glioma', 'Meningioma', 'No Tumor', 'Pituitary']
|
| 94 |
+
|
| 95 |
+
bars = ax.barh(class_labels, demo_probs, color=colors, height=0.65)
|
| 96 |
+
ax.set_xlabel('Confidence Level', fontsize=12, color='#E2E8F0')
|
| 97 |
+
ax.set_title('Demo Confidence Distribution', fontsize=14, color='#00D9FF')
|
| 98 |
+
ax.set_xlim(0, 1.0)
|
| 99 |
+
ax.set_facecolor('#1E1E1E')
|
| 100 |
+
|
| 101 |
+
for spine in ax.spines.values():
|
| 102 |
+
spine.set_color('#2D3748')
|
| 103 |
+
ax.tick_params(colors='#CBD5E0')
|
| 104 |
+
|
| 105 |
+
for i, (bar, prob) in enumerate(zip(bars, demo_probs)):
|
| 106 |
+
ax.text(prob + 0.02, i, f'{prob*100:.1f}%', va='center',
|
| 107 |
+
fontsize=11, fontweight='600', color='#00D9FF')
|
| 108 |
+
|
| 109 |
+
plt.tight_layout()
|
| 110 |
+
chart_path = '/tmp/demo_chart.png'
|
| 111 |
+
plt.savefig(chart_path, dpi=120, facecolor='#1E1E1E')
|
| 112 |
+
plt.close()
|
| 113 |
+
|
| 114 |
+
return report, None, img_rgb, chart_path
|
| 115 |
+
|
| 116 |
+
# TODO: Add actual prediction code here when models are loaded
|
| 117 |
+
|
| 118 |
+
except Exception as e:
|
| 119 |
+
error_msg = f"❌ Error: {str(e)}"
|
| 120 |
+
return error_msg, None, None, None
|
| 121 |
+
|
| 122 |
+
def generate_demo_report(model_choice, language):
|
| 123 |
+
"""Generate demo report"""
|
| 124 |
+
|
| 125 |
+
if language == 'en':
|
| 126 |
+
return f"""
|
| 127 |
+
╔══════════════════════════════════════════════════════════════╗
|
| 128 |
+
║ BRAIN TUMOR CLASSIFICATION REPORT ║
|
| 129 |
+
╚════════════════════════���═════════════════════════════════════╝
|
| 130 |
+
|
| 131 |
+
🔄 DEMO MODE
|
| 132 |
+
|
| 133 |
+
Selected Model: {model_choice}
|
| 134 |
+
Language: English
|
| 135 |
+
|
| 136 |
+
SYSTEM STATUS:
|
| 137 |
+
━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━
|
| 138 |
+
✅ Interface: Running
|
| 139 |
+
✅ Image Processing: Active
|
| 140 |
+
⚠️ AI Models: Upload .pth files to enable predictions
|
| 141 |
+
|
| 142 |
+
TO ENABLE FULL FUNCTIONALITY:
|
| 143 |
+
━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━
|
| 144 |
+
1. Upload your trained model files (.pth) to this Space
|
| 145 |
+
2. Update the load_models() function in app.py
|
| 146 |
+
3. Redeploy the Space
|
| 147 |
+
|
| 148 |
+
FEATURES READY:
|
| 149 |
+
━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━
|
| 150 |
+
✅ Multi-language support (EN/BN/ES)
|
| 151 |
+
✅ Model selection interface
|
| 152 |
+
✅ Image upload & preprocessing
|
| 153 |
+
✅ Professional UI with dark theme
|
| 154 |
+
✅ Confidence visualization
|
| 155 |
+
✅ Medical report generation
|
| 156 |
+
|
| 157 |
+
The system is fully configured and ready for your trained models!
|
| 158 |
+
|
| 159 |
+
━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━
|
| 160 |
+
⚠️ MEDICAL DISCLAIMER
|
| 161 |
+
━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━
|
| 162 |
+
This AI system is for educational and research purposes only.
|
| 163 |
+
Always consult qualified healthcare professionals for medical advice.
|
| 164 |
+
━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━
|
| 165 |
+
"""
|
| 166 |
+
elif language == 'bn':
|
| 167 |
+
return f"""
|
| 168 |
+
╔══════════════════════════════════════════════════════════════╗
|
| 169 |
+
║ মস্তিষ্ক টিউমার শ্রেণীবিভাগ রিপোর্ট ║
|
| 170 |
+
╚══════════════════════════════════════════════════════════════╝
|
| 171 |
+
|
| 172 |
+
🔄 ডেমো মোড
|
| 173 |
+
|
| 174 |
+
নির্বাচিত মডেল: {model_choice}
|
| 175 |
+
ভাষা: বাংলা
|
| 176 |
+
|
| 177 |
+
সিস্টেম স্ট্যাটাস:
|
| 178 |
+
━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━
|
| 179 |
+
✅ ইন্টারফেস: চলছে
|
| 180 |
+
✅ ছবি প্রসেসিং: সক্রিয়
|
| 181 |
+
⚠️ এআই মডেল: ভবিষ্যদ্বাণী সক্রিয় করতে .pth ফাইল আপলোড করুন
|
| 182 |
+
|
| 183 |
+
সম্পূর্ণ কার্যকারিতা সক্ষম করতে:
|
| 184 |
+
━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━
|
| 185 |
+
1. এই স্পেসে আপনার প্রশিক্ষিত মডেল ফাইল (.pth) আপলোড করুন
|
| 186 |
+
2. app.py তে load_models() ফাংশন আপডেট করুন
|
| 187 |
+
3. স্পেস পুনরায় ডিপ্লয় করুন
|
| 188 |
+
|
| 189 |
+
প্রস্তুত বৈশিষ্ট্য:
|
| 190 |
+
━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━
|
| 191 |
+
✅ বহু-ভাষা সমর্থন
|
| 192 |
+
✅ মডেল নির্বাচন ইন্টারফেস
|
| 193 |
+
✅ ছবি আপলোড এবং প্রসেসিং
|
| 194 |
+
✅ পেশাদার UI
|
| 195 |
+
✅ আত্মবিশ্বাস ভিজুয়ালাইজেশন
|
| 196 |
+
|
| 197 |
+
সিস্টেম সম্পূর্ণভাবে কনফিগার করা এবং প্রস্তুত!
|
| 198 |
+
"""
|
| 199 |
+
else: # Spanish
|
| 200 |
+
return f"""
|
| 201 |
+
╔══════════════════════════════════════════════════════════════╗
|
| 202 |
+
║ INFORME DE CLASIFICACIÓN DE TUMOR CEREBRAL ║
|
| 203 |
+
╚══════════════════════════════════════════════════════════════╝
|
| 204 |
+
|
| 205 |
+
🔄 MODO DEMO
|
| 206 |
+
|
| 207 |
+
Modelo seleccionado: {model_choice}
|
| 208 |
+
Idioma: Español
|
| 209 |
+
|
| 210 |
+
ESTADO DEL SISTEMA:
|
| 211 |
+
━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━
|
| 212 |
+
✅ Interfaz: Funcionando
|
| 213 |
+
✅ Procesamiento de imagen: Activo
|
| 214 |
+
⚠️ Modelos IA: Sube archivos .pth para habilitar predicciones
|
| 215 |
+
|
| 216 |
+
PARA HABILITAR FUNCIONALIDAD COMPLETA:
|
| 217 |
+
━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━
|
| 218 |
+
1. Sube tus archivos de modelo entrenados (.pth) a este Space
|
| 219 |
+
2. Actualiza la función load_models() en app.py
|
| 220 |
+
3. Redespliegue el Space
|
| 221 |
+
|
| 222 |
+
¡El sistema está completamente configurado y listo para tus modelos!
|
| 223 |
+
"""
|
| 224 |
+
|
| 225 |
+
# ===== GRADIO INTERFACE =====
|
| 226 |
+
css = """
|
| 227 |
+
@import url('https://fonts.googleapis.com/css2?family=Inter:wght@400;500;600;700&display=swap');
|
| 228 |
+
|
| 229 |
+
:root {
|
| 230 |
+
--bg-primary: #121212;
|
| 231 |
+
--bg-secondary: #1E1E1E;
|
| 232 |
+
--accent-cyan: #00D9FF;
|
| 233 |
+
--text-primary: #E2E8F0;
|
| 234 |
+
}
|
| 235 |
+
|
| 236 |
+
* {
|
| 237 |
+
font-family: 'Inter', sans-serif !important;
|
| 238 |
+
}
|
| 239 |
+
|
| 240 |
+
.gradio-container {
|
| 241 |
+
background: var(--bg-primary) !important;
|
| 242 |
+
max-width: 100% !important;
|
| 243 |
+
}
|
| 244 |
+
|
| 245 |
+
.gr-button-primary {
|
| 246 |
+
background: linear-gradient(135deg, var(--accent-cyan) 0%, #2196F3 100%) !important;
|
| 247 |
+
color: #000000 !important;
|
| 248 |
+
font-weight: 700 !important;
|
| 249 |
+
border: none !important;
|
| 250 |
+
}
|
| 251 |
+
|
| 252 |
+
.gr-button-primary:hover {
|
| 253 |
+
transform: translateY(-2px) !important;
|
| 254 |
+
box-shadow: 0 8px 20px rgba(0, 217, 255, 0.3) !important;
|
| 255 |
+
}
|
| 256 |
+
"""
|
| 257 |
+
|
| 258 |
+
with gr.Blocks(
|
| 259 |
+
css=css,
|
| 260 |
+
title="Brain Tumor AI Classifier - Abdullah Rubab",
|
| 261 |
+
theme=gr.themes.Base(primary_hue="cyan", secondary_hue="blue").set(
|
| 262 |
+
body_background_fill="#121212",
|
| 263 |
+
background_fill_primary="#1E1E1E",
|
| 264 |
+
border_color_primary="#2D3748"
|
| 265 |
+
)
|
| 266 |
+
) as demo:
|
| 267 |
+
|
| 268 |
+
gr.Markdown("""
|
| 269 |
+
# 🧠 Brain Tumor AI Classifier
|
| 270 |
+
|
| 271 |
+
**Advanced Deep Learning Medical Image Analysis**
|
| 272 |
+
|
| 273 |
+
*Powered by ConvNeXt | Vision Transformer | EfficientNetV2*
|
| 274 |
+
""")
|
| 275 |
+
|
| 276 |
+
with gr.Row():
|
| 277 |
+
# Left Panel
|
| 278 |
+
with gr.Column(scale=3, min_width=380):
|
| 279 |
+
with gr.Group():
|
| 280 |
+
gr.Markdown("### 📤 Upload MRI Scan")
|
| 281 |
+
image_input = gr.Image(
|
| 282 |
+
type="pil",
|
| 283 |
+
label="Brain MRI Scan",
|
| 284 |
+
height=320
|
| 285 |
+
)
|
| 286 |
+
|
| 287 |
+
language_choice = gr.Radio(
|
| 288 |
+
choices=[
|
| 289 |
+
("🇺🇸 English", "en"),
|
| 290 |
+
("🇧🇩 বাংলা", "bn"),
|
| 291 |
+
("🇪🇸 Español", "es")
|
| 292 |
+
],
|
| 293 |
+
value="en",
|
| 294 |
+
label="🌐 Language"
|
| 295 |
+
)
|
| 296 |
+
|
| 297 |
+
model_choice = gr.Dropdown(
|
| 298 |
+
choices=["Ensemble", "ConvNeXt-Base", "ViT-B/16", "EfficientNetV2-M"],
|
| 299 |
+
value="Ensemble",
|
| 300 |
+
label="🤖 AI Model"
|
| 301 |
+
)
|
| 302 |
+
|
| 303 |
+
use_tta = gr.Checkbox(
|
| 304 |
+
label="🔄 Test-Time Augmentation",
|
| 305 |
+
value=False
|
| 306 |
+
)
|
| 307 |
+
|
| 308 |
+
show_explanations = gr.Checkbox(
|
| 309 |
+
label="🔍 Grad-CAM++ Explanation",
|
| 310 |
+
value=True
|
| 311 |
+
)
|
| 312 |
+
|
| 313 |
+
predict_btn = gr.Button(
|
| 314 |
+
"🔍 ANALYZE SCAN",
|
| 315 |
+
variant="primary",
|
| 316 |
+
size="lg"
|
| 317 |
+
)
|
| 318 |
+
|
| 319 |
+
with gr.Group():
|
| 320 |
+
gr.Markdown("""
|
| 321 |
+
### 📊 Model Performance
|
| 322 |
+
|
| 323 |
+
**Ensemble:** 96%+ accuracy
|
| 324 |
+
**ConvNeXt:** 99.9% F1 score
|
| 325 |
+
**ViT:** 99%+ F1 score
|
| 326 |
+
**EfficientNetV2:** 99%+ F1 score
|
| 327 |
+
|
| 328 |
+
### 🏥 Detectable Tumors
|
| 329 |
+
|
| 330 |
+
• Glioma (গ্লাইওমা, Glioma)
|
| 331 |
+
• Meningioma (মেনিনজিওমা, Meningioma)
|
| 332 |
+
• Pituitary (পিটুইটারি, Pituitario)
|
| 333 |
+
• No Tumor (কোন টিউমার নেই, Sin tumor)
|
| 334 |
+
""")
|
| 335 |
+
|
| 336 |
+
# Right Panel
|
| 337 |
+
with gr.Column(scale=7, min_width=700):
|
| 338 |
+
with gr.Group():
|
| 339 |
+
gr.Markdown("### 📊 Analysis Results")
|
| 340 |
+
result_output = gr.Textbox(
|
| 341 |
+
label="Medical Report",
|
| 342 |
+
lines=22,
|
| 343 |
+
show_copy_button=True
|
| 344 |
+
)
|
| 345 |
+
|
| 346 |
+
with gr.Row():
|
| 347 |
+
uploaded_display = gr.Image(
|
| 348 |
+
label="📸 Uploaded Scan",
|
| 349 |
+
height=280
|
| 350 |
+
)
|
| 351 |
+
|
| 352 |
+
chart_output = gr.Image(
|
| 353 |
+
label="📊 Confidence Distribution",
|
| 354 |
+
height=280
|
| 355 |
+
)
|
| 356 |
+
|
| 357 |
+
explanation_output = gr.Image(
|
| 358 |
+
label="🔥 Grad-CAM++ Explanation",
|
| 359 |
+
height=320
|
| 360 |
+
)
|
| 361 |
+
|
| 362 |
+
# Event handler
|
| 363 |
+
predict_btn.click(
|
| 364 |
+
fn=predict_brain_tumor,
|
| 365 |
+
inputs=[image_input, model_choice, use_tta, show_explanations, language_choice],
|
| 366 |
+
outputs=[result_output, explanation_output, uploaded_display, chart_output]
|
| 367 |
+
)
|
| 368 |
+
|
| 369 |
+
# Footer
|
| 370 |
+
gr.Markdown("""
|
| 371 |
+
---
|
| 372 |
+
|
| 373 |
+
## 👨💻 Developer Information
|
| 374 |
+
|
| 375 |
+
**Abdullah Rubab**
|
| 376 |
+
|
| 377 |
+
🏫 Daffodil International University, Bangladesh
|
| 378 |
+
📧 rubab2712@gmail.com
|
| 379 |
+
🐙 GitHub: [@ABRUBAB](https://github.com/ABRUBAB)
|
| 380 |
+
🔗 Project: [brain-tumor-mri-classification](https://github.com/ABRUBAB/brain-tumor-mri-classification)
|
| 381 |
+
|
| 382 |
+
---
|
| 383 |
+
|
| 384 |
+
## ⚠️ IMPORTANT MEDICAL DISCLAIMER
|
| 385 |
+
|
| 386 |
+
This AI system is developed for **educational and research purposes only**.
|
| 387 |
+
|
| 388 |
+
- This tool should **NOT** be used as a substitute for professional medical diagnosis
|
| 389 |
+
- Always seek advice from qualified healthcare providers
|
| 390 |
+
- Never disregard professional medical advice based on AI predictions
|
| 391 |
+
- This system is not FDA/EMA approved
|
| 392 |
+
- Clinical decisions should only be made by certified medical practitioners
|
| 393 |
+
|
| 394 |
+
---
|
| 395 |
+
|
| 396 |
+
© 2025 Abdullah Rubab | Academic Research Project
|
| 397 |
+
""")
|
| 398 |
+
|
| 399 |
+
# Launch
|
| 400 |
+
if __name__ == "__main__":
|
| 401 |
+
demo.launch()
|
requirements.txt
ADDED
|
@@ -0,0 +1,30 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
# Core Deep Learning Frameworks
|
| 2 |
+
torch>=2.0.0
|
| 3 |
+
torchvision>=0.15.0
|
| 4 |
+
timm>=0.9.0
|
| 5 |
+
|
| 6 |
+
# Computer Vision & Image Processing
|
| 7 |
+
opencv-python-headless>=4.8.0
|
| 8 |
+
albumentations>=1.3.0
|
| 9 |
+
pillow>=10.0.0
|
| 10 |
+
|
| 11 |
+
# Data Science & Machine Learning
|
| 12 |
+
numpy>=1.24.0
|
| 13 |
+
pandas>=2.0.0
|
| 14 |
+
scikit-learn>=1.3.0
|
| 15 |
+
scipy>=1.11.0
|
| 16 |
+
|
| 17 |
+
# Visualization
|
| 18 |
+
matplotlib>=3.7.0
|
| 19 |
+
seaborn>=0.12.0
|
| 20 |
+
|
| 21 |
+
# Explainable AI
|
| 22 |
+
grad-cam>=1.4.0
|
| 23 |
+
pytorch-grad-cam>=1.4.0
|
| 24 |
+
|
| 25 |
+
# Web Interface
|
| 26 |
+
gradio>=4.0.0
|
| 27 |
+
|
| 28 |
+
# Utilities
|
| 29 |
+
pyyaml>=6.0
|
| 30 |
+
tqdm>=4.65.0
|