File size: 1,496 Bytes
3af943a |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 |
---
license: mit
library_name: pytorch
pipeline_tag: image-classification
base_model: vgg16
metrics:
- accuracy
tags:
- brain-tumor
- medical-imaging
- mri
- vgg16
- transfer-learning
- colorized-images
- pytorch
- image-classification
language:
- en
---
# Brain Tumor Classification using VGG16 (Colorized MRI)
This repository contains a **VGG16 transfer learning model trained on enhanced colorized MRI images** for automated brain tumor classification.
## 🧠 Tumor Classes
- Glioma
- Meningioma
- Pituitary
## 📊 Model Performance
- **Test Accuracy:** **88.70%**
- **Framework:** PyTorch
- **Architecture:** VGG16 (Transfer Learning)
- **Pre-trained on:** ImageNet
- **Input Size:** 224×224 RGB
- **Number of Classes:** 3
## 🎨 Colorization Strategy
MRI images were enhanced using **CLAHE** and converted into multiple colormap representations to study the impact of color information on classification performance.
## 🏆 Best Model Checkpoint
represents the **best-performing checkpoint**, saved at peak validation accuracy.
## 🔬 Training Highlights
- Transfer learning with frozen convolution layers
- Fine-tuned classifier head
- Data augmentation
- Stratified train/validation/test split (70/15/15)
- Early stopping and learning rate scheduling
## ⚠️ Disclaimer
This model is intended **strictly for research and educational purposes** and must not be used for clinical diagnosis or treatment planning.
## 👤 Author
**Prashant Parwani**
The uploaded file: |