Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Added Malaria Cell Classification Using CNN and Transfer Learning #982

Merged
merged 4 commits into from
Nov 5, 2024

Conversation

IkkiOcean
Copy link
Contributor

Pull Request for PyVerse 💡

Requesting to submit a pull request to the PyVerse repository.


Issue Title

Malaria Cell Classification Using CNN and Transfer Learning

  • I have provided the issue title.

Info about the Related Issue

What's the goal of the project?
The goal of this project is to automate the classification of malaria-infected and uninfected cells using machine learning and deep learning models. By utilizing CNNs and transfer learning, the project aims to increase diagnostic accuracy and reduce the time needed to identify parasitized cells from microscope images.

  • I have described the aim of the project.

Name

Vivek Prakash

  • I have provided my name.

GitHub ID

https://github.com/IkkiOcean

  • I have provided my GitHub ID.

Email ID

vivekprakash.st@gmail.com

  • I have provided my email ID.

Identify Yourself

Participating in GSSOC-EXT

  • I have mentioned my participant role.

Closes

**Closes: #926 **

  • I have provided the issue number.

Describe the Add-ons or Changes You've Made

Give a clear description of what you have added or modified.
This PR introduces the following changes:

  1. Dataset Integration: Added preprocessed malaria cell images categorized as parasitized and uninfected, including data augmentation.
  2. Models:
    • MLP: Implemented as a baseline with ~65% accuracy.
    • CNN: Custom architecture achieving 96% accuracy on training and 94% on testing.
    • CNN with Regularization: Regularized CNN with similar accuracy to base CNN.
    • Transfer Learning (VGG19): Fine-tuned VGG19 model trained for 1 epoch to leverage pre-trained weights.
  3. Documentation: Added README with project overview, dataset details, model architectures, and performance summaries.
  • I have described my changes.

Type of Change

Select the type of change:

  • Bug fix (non-breaking change which fixes an issue)
  • New feature (non-breaking change which adds functionality)
  • Code style update (formatting, local variables)
  • Breaking change (fix or feature that would cause existing functionality to not work as expected)
  • This change requires a documentation update

How Has This Been Tested?

Describe how your changes have been tested.
The models were trained and tested on the malaria cell dataset, with accuracies recorded on training and testing sets to evaluate performance. The models were also validated with data augmentation to check for overfitting and generalization. Model performance was compared through metrics such as accuracy on both train and test datasets.

  • I have described my testing process.

Checklist

Please confirm the following:

  • My code follows the guidelines of this project.
  • I have performed a self-review of my own code.
  • I have commented my code, particularly wherever it was hard to understand.
  • I have made corresponding changes to the documentation.
  • My changes generate no new warnings.
  • I have added things that prove my fix is effective or that my feature works.
  • Any dependent changes have been merged and published in downstream modules.

Copy link

👋 Thank you for opening this pull request! We're excited to review your contribution. Please give us a moment, and we'll get back to you shortly!

Feel free to join our community on Discord to discuss more!

@UTSAVS26
Copy link
Owner

UTSAVS26 commented Nov 1, 2024

@IkkiOcean explain about the dataset too about its features and parameters in the dataset readme file.

@UTSAVS26 UTSAVS26 added Contributor Denotes issues or PRs submitted by contributors to acknowledge their participation. Status: Review Ongoing 🔄 PR is currently under review and awaiting feedback from reviewers. level1 gssoc-ext hacktoberfest labels Nov 1, 2024
@IkkiOcean
Copy link
Contributor Author

@IkkiOcean explain about the dataset too about its features and parameters in the dataset readme file.

okay sure.

@IkkiOcean
Copy link
Contributor Author

@UTSAVS26 done with the required changes, please merge it soon

@UTSAVS26 UTSAVS26 merged commit 0fffe58 into UTSAVS26:main Nov 5, 2024
@UTSAVS26 UTSAVS26 added Status: Approved ✔️ PRs that have passed review and are approved for merging. and removed Status: Review Ongoing 🔄 PR is currently under review and awaiting feedback from reviewers. labels Nov 5, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
Contributor Denotes issues or PRs submitted by contributors to acknowledge their participation. gssoc-ext hacktoberfest level1 Status: Approved ✔️ PRs that have passed review and are approved for merging.
Projects
None yet
Development

Successfully merging this pull request may close these issues.

[Code Addition Request]: Malaria Cells Classification using CNN and Transfer Learning
2 participants