Skip to content

This project utilizes InceptionResNetV2 for brain tumor classification. Trained on a curated dataset, the model distinguishes between tumor and non-tumor brain images. With GPU acceleration, it ensures efficient training, and results are presented through metrics, graphs, and random image predictions. A valuable tool for medical image analysis. ๐ŸŒ๏ฟฝ

Notifications You must be signed in to change notification settings

MohammedAfthab18/Brain-Tumor-Classification-using-inceptionresnetV2

Folders and files

NameName
Last commit message
Last commit date

Latest commit

ย 

History

6 Commits
ย 
ย 
ย 
ย 

Repository files navigation

Brain Tumor Classification using InceptionResNetV2 ๐Ÿง ๐Ÿ”

Overview

This project focuses on the classification of brain tumor images using the InceptionResNetV2 deep learning architecture. The goal is to develop an accurate model capable of distinguishing between tumor and non-tumor images.

Project Highlights ๐Ÿš€

  • Dataset Description: Brief overview of the dataset, including class distribution.
  • Data Generators: Efficient loading and augmentation of data during training, testing, and validation.
  • Class Labels: 'Tumor': 0, 'non-Tumor': 1.

Dataset Statistics ๐Ÿ“Š

  • Training Dataset:
    • Number of datasets : 98

GPU and Model Checkpoint ๐Ÿ–ฅ๏ธ๐Ÿ’พ

  • Check GPU Availability: Ensure efficient training with GPU acceleration.
  • Model Checkpoint: Save the model during training for future use.

Model Architecture ๐Ÿง ๐Ÿ—๏ธ

  • InceptionResNet for Feature Extractor: Utilized the power of InceptionResNetV2 for automatic feature extraction.

Model Training ๐Ÿš‚๐Ÿ’จ

  • Training Starts Here: Begin the training process.
  • Save Model History: Save training history in a CSV file for analysis.
  • Accuracy and Loss Graphs: Visualize the training and validation performance.

Model Evaluation ๐Ÿง๐Ÿ“ˆ

  • Testing Starts Here: Evaluate the trained model on the test set.
  • Evaluation Metrics: Compute accuracy, precision, recall, and F1-score.
  • Classification Report: Detailed report of model performance.
  • Confusion Matrix: Visualize model predictions.
  • ROC Curve: Evaluate model performance.

Model Predictions ๐Ÿ“ธ๐Ÿ”ฎ

  • Random Image Predictions: Check the model's predictions on random images from the test dataset.

About

This project utilizes InceptionResNetV2 for brain tumor classification. Trained on a curated dataset, the model distinguishes between tumor and non-tumor brain images. With GPU acceleration, it ensures efficient training, and results are presented through metrics, graphs, and random image predictions. A valuable tool for medical image analysis. ๐ŸŒ๏ฟฝ

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published