Giter VIP home page Giter VIP logo

mohammad95labbaf / brain-tumor-transferlearning Goto Github PK

View Code? Open in Web Editor NEW
0.0 1.0 0.0 764 KB

The Brain Tumor MRI Dataset from Kaggle is employed for automated brain tumor detection and classification research. Investigated methods include using pre-trained models (VGG16, ResNet50, and ViT). 🧠🔍

Jupyter Notebook 100.00%
cnn cnn-keras deep-learning deep-neural-networks deeplearning keras keras-tensorflow neural-network neural-networks pretrained-models resnet-50 transfer-learning tumor-classification tumor-detection vgg16 vision-transformer vit kaggle-dataset vision

brain-tumor-transferlearning's Introduction

Brain-Tumor-TransferLearning

Dataset:

  • The dataset utilized for this study is the Brain Tumor MRI Dataset sourced from Kaggle.
  • It consists of a carefully curated collection of brain MRI scans specifically chosen to facilitate research in automated brain tumor detection and classification using the Keras library.
  • The dataset aims to enhance diagnostic accuracy and includes a randomized subset (20% of the original data) categorized into 'yes' (tumor present) and 'no' (healthy) tumor classes for both training and validation purposes.

Investigated Approaches:

  1. VGG16 + FC (VGG16 with Fully Connected Layers):

    • VGG16 is a deep CNN architecture comprising 16 weight layers.
    • It was pre-trained on the ImageNet dataset.
    • Additional fully connected layers are incorporated for classification.
  2. VGG + CNN2D (VGG16 with Additional Convolutional Layers):

    • This approach extends VGG16 by introducing more convolutional layers for enhanced feature extraction.
  3. ResNet50 + FC (ResNet50 with Fully Connected Layers):

    • ResNet50, a deep residual network with 50 layers, addresses the vanishing gradient problem using skip connections.
    • Fully connected layers are added for classification.
  4. ResNet50 + CNN2D (ResNet50 with Additional Convolutional Layers):

    • Building upon ResNet50, this variant incorporates additional convolutional layers.
  5. ViT (Vision Transformer) + FC (Fully Connected Layers):

    • Originally designed for natural language processing, ViT is a transformer-based architecture.
    • It has been adapted for image classification by treating images as sequences of patches.
    • Fully connected layers are employed for classification⁷.

Investigation Objectives:

  • Evaluate the effectiveness of transfer learning using pre-trained models (VGG16, ResNet50, and ViT) for brain tumor classification.
  • Compare the performance of different architectures in terms of accuracy, sensitivity, specificity, and other relevant metrics.
  • Gain insights into how transfer learning impacts model convergence, generalization, and robustness.

Instructions for Kaggle API

  1. Download Kaggle API:

    • Install the Kaggle API by running pip install kaggle.
  2. Kaggle API Token:

    • Go to your Kaggle account settings and generate an API token.
    • Save the token as kaggle.json in the root directory of this repository.
  3. Download Dataset:

    • Use the Kaggle API to download the dataset:
      kaggle datasets download -d  preetviradiya/brian-tumor-dataset
      
  4. Upload Kaggle API Token to Colab/Notebook:

    • If using Colab or Jupyter Notebook, upload the kaggle.json token to your environment.
    • Use the following code snippet:
      from google.colab import files
      files.upload()
  5. Unzip Dataset:

    • Unzip the downloaded dataset:
      unzip brian-tumor-dataset.zip
      

brain-tumor-transferlearning's People

Contributors

mohammad95labbaf avatar

Watchers

 avatar

Recommend Projects

  • React photo React

    A declarative, efficient, and flexible JavaScript library for building user interfaces.

  • Vue.js photo Vue.js

    🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.

  • Typescript photo Typescript

    TypeScript is a superset of JavaScript that compiles to clean JavaScript output.

  • TensorFlow photo TensorFlow

    An Open Source Machine Learning Framework for Everyone

  • Django photo Django

    The Web framework for perfectionists with deadlines.

  • D3 photo D3

    Bring data to life with SVG, Canvas and HTML. 📊📈🎉

Recommend Topics

  • javascript

    JavaScript (JS) is a lightweight interpreted programming language with first-class functions.

  • web

    Some thing interesting about web. New door for the world.

  • server

    A server is a program made to process requests and deliver data to clients.

  • Machine learning

    Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.

  • Game

    Some thing interesting about game, make everyone happy.

Recommend Org

  • Facebook photo Facebook

    We are working to build community through open source technology. NB: members must have two-factor auth.

  • Microsoft photo Microsoft

    Open source projects and samples from Microsoft.

  • Google photo Google

    Google ❤️ Open Source for everyone.

  • D3 photo D3

    Data-Driven Documents codes.