Skip to content

AI-MOO/Brain_Tumor_Detector_Deep_Learning

Repository files navigation

Brain Tumors Detection Using Deep Learning 🧠


Introduction

This repository contains a deep learning model based on a convolutional neural network (CNN) used to detect brain tumors from MRI images.

There are two pre-trained models in this repo :

  1. Binary Cross entropy : BrainTumor10Epochs.h5
  2. Categorical Cross Entropy : BrainTumorCategorical10Epochs.h5

Before you run the model

  1. Clone the repository recursively:

    • git clone --recurse-submodules https://github.com/AI-MOO/Brain_Tumor_Detector_Deep_Learning.git
  2. Make sure that you fulfill all the requirements: Python 3.6.8 or later with all packages in requirements.txt

  3. Run the model through flask application module : app.py in your terminal

    python app.py
    

Dataset

  1. Br35H : Brain Tumor Detection 2020
  2. to train your custom model use main_train.py module

Credit & Tutorial