SartajBhuvaji / Brain-Tumor-Classification-Using-Deep-Learning-Algorithms

To Detect and Classify Brain Tumors using CNN and ANN as an asset of Deep Learning and to examine the position of the tumor.
https://sartajbhuvaji.github.io/Brain-Tumor-Classification-Website/
43 stars 20 forks source link
cnn imageclassification machine-learning neural-network tensorflow transferlearning


logo

Brain Tumor Classification Using Deep Learning Algorithms

SPPU
Sartaj Bhuvaji · Ankita Kadam · Prajakta Bhumkar . Sameer Dedge

Problem Definition To Detect and Classify Brain Tumors using CNN and TL.

[![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/drive/1yHKKk8I4QfKc1qv2cXCNbgIiUtBRk1rF?usp=sharing)

Abstract

A brain tumor is one aggressive disease. More than 84,000 people will receive a primary brain tumor diagnosis in 2021 and an estimated 18,600 people will die from a malignant brain tumor (brain cancer) in 2021.[8] The best technique to detect brain tumors is by using Magnetic Resonance Imaging (MRI). More than any other cancer, brain tumors can have lasting and life-altering physical, cognitive, and psychological impacts on a patient’s life and hence faster diagnosis and best treatment plan should be devised to improve the life expectancy and well-being of these patients. Neural networks have shown colossal accuracy in image classification and segmentation problems.

In this project, we propose comparative studies of various deep learning models based on different types of Neural Networks (ANN, CNN, TL) to first identify brain tumors and then classify them into Benign Tumors, Malignant Tumors, or Pituitary Tumors. The data set holds 3260 images on T1-weighted contrast-enhanced images that were cleaned and augmented. The best ANN model concluded with an accuracy of 78% and the best CNN model consisting of 3 convolution layers had an accuracy of 90%. The VGG16 (retrained on the dataset) model surpasses other ANN, CNN, and TL models for multi-class tumor classification. This proposed network performs significantly better with a validation accuracy of 94% and an F1-Score of 91.

## Demo

logo
## DataSet Dataset: [Kaggle](https://www.kaggle.com/datasets/sartajbhuvaji/brain-tumor-classification-mri)
dataset_description
## Contributing to the project: * Fork the [Master Branch](https://github.com/SartajBhuvaji/Brain-Tumor-Classification-Using-Deep-Learning-Algorithms) * Clone your branch * If you want to contribute by adding your classification model, create a new folder inside ```Contributions/Your_Name``` * Make sure to add a detailed README File. ## Related Repositories: * [Data Set](https://github.com/SartajBhuvaji/Brain-Tumor-Classification-DataSet/) * [HuggingFace](https://huggingface.co/datasets/sartajbhuvaji/Brain-Tumor-Classification) * [Website](https://sartajbhuvaji.github.io/Brain-Tumor-Classification-Website/) * [Primary Repo](https://github.com/SartajBhuvaji/Brain-Tumor-Classification-Using-Deep-Learning-Algorithms/tree/master) * [Runner](https://github.com/SartajBhuvaji/Brain-Tumor-Classification-Using-Deep-Learning-Algorithms-Runner) * [Kaggle](https://www.kaggle.com/datasets/sartajbhuvaji/brain-tumor-classification-mri) * [Medium Article](https://medium.com/@sartajbhuvaji/brain-tumor-classification-546a72d4103b) ## Other Collaborators * [Aditya Patil](https://github.com/Adityathere) * [Fatih Sen](https://github.com/fatihsen20) * [Vishwesh Salodkar](https://github.com/VishweshSalodkar) ##
[![Python 3.7](https://img.shields.io/badge/python-3.7-blue.svg)](https://www.python.org/downloads/release/python-370/)