A deep learning project for automatic brain tumor classification from MRI images using a hybrid model that combines the strengths of multiple architectures for improved accuracy and generalization.
This project was developed and trained in Google Colab, and can be run end-to-end in the provided .ipynb notebook.
- π₯ Medical Imaging Focus β Designed for classifying brain MRI scans into tumor categories.
- π¬ Hybrid Deep Learning Architecture β Combines convolutional neural networks (CNNs) with additional machine learning layers for improved performance.
- π Training and Evaluation β Includes preprocessing, model training, validation, and performance metrics.
- β Google Colab Ready β Fully runnable without local setup.
- π Visualization Tools β Plots for accuracy, loss curves, and confusion matrices.
brain-tumor-classification/
β
βββ brain-tumor-classification-hybrid-model.ipynb # Main Google Colab notebook
βββ README.md # Project documentation
Click the badge below to open the notebook in Colab:
You can use publicly available datasets such as:
Place the dataset in the appropriate directory or mount Google Drive in Colab.
If using Colab, dependencies are installed within the notebook automatically.
Execute all cells in the brain-tumor-classification-hybrid-model.ipynb notebook to train and evaluate the model.
- Feature Extractor β A CNN backbone for capturing spatial features from MRI images.
- Hybrid Layer β Combines CNN output with a secondary model (e.g., Dense layers, classical ML classifier) for better classification accuracy.
- Output Layer β Softmax activation for multi-class classification.
The notebook includes:
- Accuracy and loss curves over training epochs
- Confusion matrix for detailed class-wise performance
- Example predictions with true vs predicted labels
- Experiment with different CNN backbones (ResNet, EfficientNet, etc.)
- Implement data augmentation for robustness
- Deploy the model via a web app (e.g., Streamlit or Flask)
This project is licensed under the MIT License β see the LICENSE file for details.
- Dataset providers (Kaggle, Figshare, etc.)
- TensorFlow / PyTorch community
- Google Colab for cloud-based development
Built to assist in medical image classification research and education.