-
Notifications
You must be signed in to change notification settings - Fork 208
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Merge pull request #982 from IkkiOcean/malaria
Added Malaria Cell Classification Using CNN and Transfer Learning
- Loading branch information
Showing
7 changed files
with
84 additions
and
0 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,28 @@ | ||
# Malaria Cell Images Dataset | ||
|
||
This dataset contains cell images for detecting malaria in human blood samples. It can be used for building and testing machine learning and deep learning models that classify cell images as either infected or uninfected with malaria parasites. | ||
|
||
**Dataset Link:** [Kaggle - Malaria Cell Images Dataset](https://www.kaggle.com/datasets/iarunava/cell-images-for-detecting-malaria) | ||
|
||
## Dataset Structure | ||
|
||
The dataset is organized into two folders, each containing labeled images for malaria-infected and uninfected cells: | ||
|
||
- **Infected:** Contains 13,800 images of cells infected with malaria. | ||
- **Uninfected:** Contains 13,758 images of cells that are uninfected. | ||
|
||
**Total Images:** 27,558 | ||
|
||
## Folder Structure | ||
``` | ||
malaria_cell_images/ | ||
├── Infected/ | ||
│ ├── C1_thinF_IMG_20150604_104722a_cell_180.png | ||
│ ├── C1_thinF_IMG_20150604_104722a_cell_181.png | ||
│ └── ... | ||
└── Uninfected/ | ||
├── C2_thinF_IMG_20150604_104722b_cell_99.png | ||
├── C2_thinF_IMG_20150604_104722b_cell_100.png | ||
└── ... | ||
``` | ||
|
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
1 change: 1 addition & 0 deletions
1
Deep_Learning/Malaria-Classification/model/malaria-classification.ipynb
Large diffs are not rendered by default.
Oops, something went wrong.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,48 @@ | ||
# Malaria Cell Classification | ||
|
||
This project involves classifying malaria-infected and uninfected cells using different machine learning and deep learning approaches. The primary goal is to automate the identification of parasitized and uninfected cells using microscopic images. | ||
|
||
## Overview | ||
|
||
Malaria is a critical disease caused by parasites, and detecting it in cells is a time-consuming process for healthcare workers. This project aims to automate the process by using machine learning and deep learning models to classify cells as parasitized or uninfected from microscope images. | ||
|
||
## Dataset | ||
|
||
- [Dataset](https://www.kaggle.com/datasets/iarunava/cell-images-for-detecting-malaria) | ||
## Models | ||
|
||
### MLP | ||
- **Performance**: | ||
- Accuracy on training data: ~65% | ||
- Accuracy on test data: ~65% | ||
|
||
While the model performs consistently on both train and test sets, the accuracy is too low to depend on for classification. | ||
|
||
### CNN | ||
- **Performance**: | ||
- Accuracy on training data: 96% | ||
- Accuracy on test data: 94% | ||
|
||
The CNN model performs well with high accuracy on both training and test datasets. | ||
|
||
### CNN with Regularization | ||
- **Performance**: | ||
- Accuracy on training data: 96% | ||
- Accuracy on test data: 94% | ||
|
||
Adding regularization to the CNN model had no significant effect on the performance, resulting in similar accuracy to the base CNN model. | ||
|
||
### Hyperparameter Tuning | ||
- **Performance**: | ||
- Improved model performance, but hyperparameter tuning was time-consuming due to the large number of permutations tried. It was challenging to achieve faster training times with exhaustive tuning. | ||
|
||
### Transfer Learning (VGG19) | ||
- **Performance**: | ||
- Using transfer learning with the VGG19 architecture was applied for 1 epoch due to high computational cost. | ||
- The model shows promise, but training for more epochs is needed for better performance. | ||
|
||
## Signature | ||
|
||
Vivek Prakash | ||
|
||
[Linkedin](https://www.linkedin.com/in/vivek-prakash-b46830283/) |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,5 @@ | ||
numpy==1.25.3 | ||
pandas==2.0.3 | ||
matplotlib==3.8.1 | ||
tensorflow==2.15.0 | ||
scikit-learn==1.3.2 |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters