Skip to content

Sedighe-Raeisi/Brain-tumor-detection

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

10 Commits
 
 
 
 
 
 

Repository files navigation

Brain Tumor Detection

In this project I used a dataset of brain MRI images. This dataset contained two classes, one class belongs to images which are healthy brains. The other class belongs to images of brains with tumor.
image image

I dowloaded the dataset from link bellow:
adress: https://www.kaggle.com/navoneel/brain-mri-images-for-brain-tumor-detection
The class of healthy brains contained 154 images.
The class of images with tumor contained 155 images.
image

I used a CNN neural network to make a binary classifier. At first, I trained the model for 10 epochs: image

and I fined tuned it for 30 more epochs.
image

I reached the value accuracy=0.92

#Model with Augmented Images

I augmented images by:

  • rotating
  • width and height shifting
  • horizontal fliping
  • shearing
  • zooming
    An example of this augmentation is placed bellow:

image image

image image

Then I repeated the model training:

image

Use a Pretrained Architecture VGG16

I used a pretrained model, VGG16, I freezed it and then I added one dense layer and a sigmoid classifier layer. I trined it for 100 epochs.

image

The best achieved accuracy for validation during training was 0.94