This repository contains the implementation of VGG16 and ResNet-50 models, techniques such as fine-tuning with layer freezing and unfreezing were applied.
Regularization methods like L1 and L2, dropout, and early stopping were employed to prevent overfitting.
Strategies such as data augmentation, transfer learning and one-hot encoding were integrated, and optimization was carried out using algorithms like Adam.
This project focuses on developing a classification model to detect brain diseases, such as cancer, tumors, and aneurysms, from CT brain scans.
The dataset is organized into two main folders:
- JPEG Images (.jpg) 📷: Contains CT brain scan images in JPEG format.
- DICOM Files (.dcm) 📁: Contains the same CT scans but in Digital Imaging and Communications in Medicine (DICOM) format.
To run this project, please follow these steps:
-
Download the dataset:
- Visit the dataset page on Kaggle: CT Scan Images of the Brain and download it.
-
Unzip and overwrite the "archive" file:
- Unzip the downloaded ZIP file.
- Locate the file named "archive" 🗃️ and overwrite this project's file.
- If running locally, ensure the dataset is in the project directory.
-
If using Google Colab, upload the dataset to your Google Drive.
-
Modify the
base_dir
in the code to point to the Colab Notebooks📂 directory:base_dir = "/content/drive/MyDrive/Colab Notebooks/"
-
Clone this repository:
git clone https://github.com/AlejandroDavidArzolaSaavedra/CNN-CT-BRAIN.git
-
Explore the dataset folders to understand the structure.
-
Use the data to train and test your models and contribute to advancing medical image analysis.