This project focuses on the diagnosis of brain tumors utilizing Convolutional Neural Networks (CNNs). The primary objective is to develop a model capable of accurately classifying brain tumors from MRI images.
The dataset employed in this study is the Brain Tumor MRI Dataset available on Kaggle. It comprises MRI scans categorized into:
- Glioma Tumor
- Meningioma Tumor
- Pituitary Tumor
- No Tumor
For this project, a subset of the dataset was used, consisting of 200 images for each type of tumor in v1 results and 800 images in v2 results.
The complete dataset is available for download at the following link: Brain Tumor MRI Dataset.
-
Data Preprocessing: MRI images are preprocessed to enhance quality and ensure consistency, including steps like skull stripping, histogram equalization, and noise reduction.
-
Model Architecture: A CNN model is constructed to extract features and perform classification. The architecture includes convolutional layers, pooling layers, and fully connected layers optimized for image classification tasks.
-
Training and Evaluation: The model is trained on the preprocessed subset of the dataset, and its performance is evaluated using metrics such as accuracy, precision, recall, and F1-score.
The CNN model’s performance was affected by the limited number of images used in training and testing. While the model initially showed potential, the reduced dataset size likely contributed to lower accuracy and generalization. Future improvements can be achieved by utilizing the full dataset and incorporating data augmentation techniques.
Follow the steps below to run .ipynb
files locally using Jupyter and Poetry on different operating systems.
Ensure that you have Poetry installed on your system. If not, install it by following the official Poetry installation guide.
- Open a terminal or command prompt.
- Create a new Poetry project or navigate to your existing project directory.
poetry init
- Add Jupyter as a dependency:
poetry add notebook
- Add TensorFlow as a dependency:
poetry add tensorflow
- Activate the virtual environment created by Poetry:
poetry shell
- Start the Jupyter Notebook server:
jupyter notebook
- Open your browser and navigate to the provided URL to access Jupyter Notebook.
- Ensure that Python and Poetry are installed and added to your PATH.
- Follow the general setup steps above.
- Use Homebrew to install Python if necessary:
brew install python
. - Follow the general setup steps above.
- Use your package manager (e.g.,
apt
,yum
, ordnf
) to install Python if necessary. - Follow the general setup steps above.
- If you encounter issues with Jupyter Notebook not being recognized, ensure the Poetry environment is active (
poetry shell
). - To stop the server, press
Ctrl+C
in the terminal where the Jupyter server is running. - TensorFlow may require specific hardware or drivers for GPU support. Refer to the official TensorFlow installation guide if you are using a GPU.