Brain tumor detection using a CNN

The objective is to build an AI capable of discerning if a brain x-rays plate has tumor or not.

View Demo · Report Bug

About The Project

Co-author: @giuseppebrb

The project aim is to use TensorFlow to train a convolutional Neural Network using a dataset of ~4900 labelled images of brain scans.
Subsequently, we will evaluate the accuracy of our model against a test dataset

This is part 1 of the final project for the executive path in "Data Science and Business Analytics" held by POLIMI Graduate School of Management of Milan ( You can find part 2 here )

Built With

• TensorFlow
• Python
• Numpy
• Jupyter Notebook
• MobileNetV2

Getting Started

Installation

You can run the project in Google Collab

• Train notebook
• Predict notebook

Or run it locally following this steps:

1. Install Python
2. Clone the repo

   git clone https://github.com/TomMenga/brain-tumor-AI-CNN.git

3. Install Jupyter notebook
4. Install the dependencies

   pip install -r requirements.txt

5. Run Jupyter

   jupyter notebook

(back to top)

Contact

Tommaso Menga - [email protected] - GitHub - Linkedin

(back to top)