This is a simple Convolutional Neural Network Model for Brain Tumor Classification into four tumor types and one control group.
- Python 2.7.x or Python 3.5.x or larger
- Tensorflow v1.0+ and Tensorflow GPU version to run on a graphic card
- Scikit learn libraries
- Numpy and Scipy
- TFLearn (http://tflearn.org/installation/)
Installation of Tensorflow and other libraries can be found in their websites.
The dataset used for this research project is available under the link https://www.dropbox.com/sh/qor3axvczfd02xy/AABtBMWs8s-_BkIFquOC5Pbua?dl=0
-
Make sure all the libaries are installed
-
Get the dataset in the same folder as the program
-
Run the proposed model
$ python proposed_model.py
You can use the proposed model, change the parameters in it to test for yourown dataset.