Skip to content

Commit

Permalink
Merge pull request #414 from IkkiOcean/malaria
Browse files Browse the repository at this point in the history
Malaria Cell Classification using CNN, and Transfer Learning (VGG19)
  • Loading branch information
UppuluriKalyani authored Oct 17, 2024
2 parents f2f551b + 5959fda commit 3fc3b8a
Show file tree
Hide file tree
Showing 5 changed files with 50 additions and 0 deletions.
1 change: 1 addition & 0 deletions Neural Networks/malaria classification/dataset/readme.md
Original file line number Diff line number Diff line change
@@ -0,0 +1 @@
Link for the dataset : https://www.kaggle.com/datasets/iarunava/cell-images-for-detecting-malaria
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.

Large diffs are not rendered by default.

48 changes: 48 additions & 0 deletions Neural Networks/malaria classification/readme.md
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/)

0 comments on commit 3fc3b8a

Please sign in to comment.