Skip to content
New issue

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

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

Already on GitHub? # to your account

PR for Hair classifier model #1012

Open
wants to merge 12 commits into
base: main
Choose a base branch
from

Conversation

KrishnaChaitanya16
Copy link

Pull Request for DL-Simplified 💡

Issue Title : Hair Type Classification using Deep Learning v1

  • Info about the related issue (Aim of the project) : Implementing an ensemble of three deep learning models: ResNet50, Convnext, and VGG16 For Hair classification .
  • Name: Krishna Chaitanya
  • GitHub ID: KrishnaChaitanya16
  • Email ID: 16krish2005@gmail.com
  • Idenitfy yourself: WOC 2025

Closes: #932

Describe the add-ons or changes you've made 📃

I have trained three models: ResNet50, VGG16, and ConvNext-Tiny, and achieved good accuracy scores in hair classification. I have also included a small demo where it captures your live webcam and predicts your hair type.

Type of change ☑️

What sort of change have you made:

  • 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? ⚙️

This has been tested with live web cam prediction, and also many pictures apart from training data , also include accuracy scores in readme file.

Checklist: ☑️

  • 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

Our team will soon review your PR. Thanks @KrishnaChaitanya16 :)

Copy link
Owner

@abhisheks008 abhisheks008 left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Please follow the project structure and put your files as per the project structure of this repository.

Project structure to be followed:

Project Folder
|- Dataset
   |- dataset.csv (dataset used for the particula project)
   |- README.md (brief about the dataset)
|- Images
   |- img1.png
   |- img2.png
   |- img3.png
|- Model
   |- project_folder.ipynb
   |- README.md
|- requirements.txt

@KrishnaChaitanya16

@abhisheks008 abhisheks008 added Status: Requested Changes Changes requested. WoC 4.0 Winter of Code 4.0 by GDG IIITK labels Feb 24, 2025
@KrishnaChaitanya16
Copy link
Author

@abhisheks008 I am facing difficulty in pushing the dataset , but github is not allowing it due to size of the dataset , i tried using git LFS still it isn't working. Any advice?

@KrishnaChaitanya16
Copy link
Author

@abhisheks008 I have changed the file structure as required , can you a look at it ?

# for free to join this conversation on GitHub. Already have an account? # to comment
Labels
Status: Requested Changes Changes requested. WoC 4.0 Winter of Code 4.0 by GDG IIITK
Projects
None yet
Development

Successfully merging this pull request may close these issues.

Hair Type Classification using Deep Learning v1
2 participants