Ajay007001/Brain-Stroke-Prediction

🧠 Brain Stroke Classification using VGG19 Transfer Learning

A high-accuracy image classification model trained on medical brain scans using VGG19 (ImageNet pretrained) to distinguish different stroke types. Achieves 89% accuracy on the Teknofest 2021 Brain Stroke dataset.

---

🚀 Highlights

• ✅ Transfer learning with VGG19
• 🧠 Medical image classification for stroke diagnosis
• 📊 Accuracy: 89%, F1-Score: 88%
• 🔬 Trained on colorized brain stroke CT/MRI images
• 📁 Dataset: Teknofest 2021, Kaggle

---

📂 Dataset

• Source: Teknofest 2021 Brain Stroke Dataset
• Type: Colorized medical images
• Split: train/, test/ with multiple stroke classes

---

🏗️ Model Details

• Base: VGG19 with frozen early layers
• Custom Head: Dense, Dropout, Softmax
• Optimizer: Adam | Loss: Sparse Categorical Crossentropy
• Input Size: 250x250 | Framework: Keras + TensorFlow 2.x

---

📈 Performance Metrics

Metric	Value
Accuracy	89%
F1-Score	88%
Precision	87%
Recall	88%

---

🧪 Usage

This model is ready for integration into clinical AI pipelines or academic research.
To load the .h5 file and run predictions, refer to example notebook ↗ (see predict.py or notebooks/).

---

📋 License

MIT License – free to use, modify, and distribute.

---

👤 Author

Ajay Vasan S
Machine Learning Engineer
🔗 GitHub | 📧 LinkedIn

---

📂 GitHub Project: AjayVasan/Brain-Stroke-Predictor

---

⭐ Star this repo if it helped you — and feel free to open issues for feedback!