Brain Tumor MRI Classification

Project information

  • Category: Deep Learning
  • Project date: 2022
  • GitHub Repo

"CS 454 - Introduction to Machine Learning and Artificial Neural Networks" course project.

Comparison of two different brain tumor MRI classification methods using VGG16 CNN model;

Case 1 - VGG16 model is trained with the original image
Case 2 - Images are passed to the Auto Encoder first, and then VGG16 model is fed with the output of the Auto Encoder

Results showed that the accuracy was increased from around 92% to around 97%, and also the training time of the VGG16 model has decreased to half when the Auto Encoder is used.

Download

Project Report

Presentation Slides

Designed by BootstrapMade