This is a simple Convolutional Neural Network Model for Brain Tumor Classification into five tumour types.
- 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://figshare.com/articles/brain_tumor_dataset/1512427/5
It can be converted to images by using Matlab and running script data_processing/dataset_mat_to_img.m
.
-
Make sure all the libaries are installed
2.1 Modify the marked lines at create_dataset.py
to include the paths to your image data
2.2 then run it to convert your image dataset to a .pkl file.
- Run the proposed model
$ python proposed_model.py my_dataset.pkl
You can use file proposed_model.py to change the parameters in it to test for yourown dataset.