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 :
- Binary Cross entropy :
BrainTumor10Epochs.h5
- Categorical Cross Entropy :
BrainTumorCategorical10Epochs.h5
-
Clone the repository recursively:
git clone --recurse-submodules https://github.com/AI-MOO/Brain_Tumor_Detector_Deep_Learning.git
-
Make sure that you fulfill all the requirements: Python 3.6.8 or later with all packages in
requirements.txt
-
Run the model through flask application module :
app.py
in your terminalpython app.py
- Br35H : Brain Tumor Detection 2020
-
Extract the images in
datasets
folder -
pip install -r requirements.txt
- to train your custom model use
main_train.py
module