Skip to content

BASED ON BRAIN MRI IMAGES DATASET WE NEED CLASSIFY THE BRAIN TUMOUR

Notifications You must be signed in to change notification settings

PawarMukesh/Brain-Tumor-Detection

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

5 Commits
 
 
 
 

Repository files navigation

BRAIN TUMOUR DETECTION

BUSINESS CASE: BASED ON BRAIN MRI IMAGES DATASET WE NEED PREDICT BRAIN TUMOUR

TASK: BINARY CLASSIFICATION

image

DATA SUMMARY

Total 274 Jpg Images are present In two classes

  • No ----> 119
  • Yes ----> 155

DEVICE THE PROJECT INTO MULTIPLE STEPS

  • Prepare training, validation and testing set
  • Get all classes labels
  • Visualise training images
  • Used CNN & VGG19 model
  • Model Compilation
  • Model Training
  • Model Evaluation
  • Model Saving
  • Prediction on test data

Architecture of prepare CNN model

image

Compile & Train Model

  • Use binary crossentropy loss with adam optimizer and metrics is accuracy
  • Balance the class weights
  • Train model of 80 epoch and use model checkpoint to save the best model
  • Plotting training, validation loss as well as accuracy

image

image

Achieved 88.24% testing accuracy

Used VGG19 transfer learning technique to get better accuracy

  • Import vgg19 library and set input image size & used imagnet dataset weight as well as not include fully connected layer at top
  • Freeze the existing weights
  • Add more layers with sigmoid activation function

Architecture of VGG19 model

image

Compile & Train model

  • Use binary crossentropy loss with adam optimizer and metrics is accuracy
  • Train model of 60 epoch and set the class weight
  • Plotting training, validation loss as well as accuracy

image

image

Save model

Achieved 92.16% testing accuracy

About

BASED ON BRAIN MRI IMAGES DATASET WE NEED CLASSIFY THE BRAIN TUMOUR

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published