Skip to content

Latest commit

 

History

History
57 lines (34 loc) · 2.35 KB

README.md

File metadata and controls

57 lines (34 loc) · 2.35 KB

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