Skip to content

Classification of Brain MRI scans into 4 categories of tumor using EfficientNet model.

Notifications You must be signed in to change notification settings

kvarun07/brain-tumor-classifier

Repository files navigation

Brain Tumor Classifier

This repository contains project on Brain Tumor Classification.

Task

Given MRI scans of the brain, the goal is to identify the type of brain tumor.

The images belong to the following 4 classes:

  1. Glioma Tumor
  2. Meningioma Tumor
  3. No Tumor
  4. Pituitary Tumor

In essence, this is an image classification problem.

Samples from Tumor Classes

alt text

Approach

EfficientNet is an efficent model that reaches State-of-the-Art accuracy on image classification transfer learning tasks. In this problem, EfficientNet with weights pre-trained on imagenet are used.

Model training curves

The performance metrics of the model with respect to number of epochs can be visualised as follows: alt text

Result

The model yielded > 95% accuracy when tested a large number of times.

About

Classification of Brain MRI scans into 4 categories of tumor using EfficientNet model.

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published