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

Hair Type Classification using Deep Learning v1 #932

Open
cosmicishan opened this issue Oct 19, 2024 · 11 comments · May be fixed by #1012
Open

Hair Type Classification using Deep Learning v1 #932

cosmicishan opened this issue Oct 19, 2024 · 11 comments · May be fixed by #1012
Assignees
Labels
Status: Assigned Assigned issue. WoC 4.0 Winter of Code 4.0 by GDG IIITK

Comments

@cosmicishan
Copy link

Deep Learning Simplified Repository (Proposing new issue)

🔴 Project Title : Hair Type Classification using Deep Learning
🔴 Aim : To build an model for classifying various hair types
🔴 Dataset : https://www.kaggle.com/datasets/kavyasreeb/hair-type-dataset/data
🔴 Approach : To enhance the classification accuracy, I propose building an ensemble model by combining predictions from:

Custom CNN
VGGNet
ResNet


📍 Follow the Guidelines to Contribute in the Project :

  • You need to create a separate folder named as the Project Title.
  • Inside that folder, there will be four main components.
    • Images - To store the required images.
    • Dataset - To store the dataset or, information/source about the dataset.
    • Model - To store the machine learning model you've created using the dataset.
    • requirements.txt - This file will contain the required packages/libraries to run the project in other machines.
  • Inside the Model folder, the README.md file must be filled up properly, with proper visualizations and conclusions.

🔴🟡 Points to Note :

  • The issues will be assigned on a first come first serve basis, 1 Issue == 1 PR.
  • "Issue Title" and "PR Title should be the same. Include issue number along with it.
  • Follow Contributing Guidelines & Code of Conduct before start Contributing.

To be Mentioned while taking the issue :


Happy Contributing 🚀

All the best. Enjoy your open source journey ahead. 😎

Copy link

Thank you for creating this issue! We'll look into it as soon as possible. Your contributions are highly appreciated! 😊

@abhisheks008
Copy link
Owner

Assigning this issue to you @cosmicishan

@abhisheks008 abhisheks008 added Status: Up for Grabs Up for grabs issue. ieee-igdtuw IEEE IGDTUW Open Source Week 2024 and removed Status: Assigned Assigned issue. level 2 Level 2 for GSSOC hacktoberfest gssoc-ext labels Nov 10, 2024
@abhisheks008 abhisheks008 removed the ieee-igdtuw IEEE IGDTUW Open Source Week 2024 label Nov 19, 2024
@abhisheks008 abhisheks008 added the WoC 4.0 Winter of Code 4.0 by GDG IIITK label Jan 1, 2025
@Rangan2005
Copy link

@abhisheks008 I have sent the proposal for this issue under WOC in devfolio
Pls have a look at it

@abhisheks008
Copy link
Owner

Proposals are under review. @Rangan2005

@abhisheks008 abhisheks008 changed the title Hair Type Classification using Deep Learning Hair Type Classification using Deep Learning v1 Jan 26, 2025
@KrishnaChaitanya16
Copy link

For issue #932 (Hair Type Classification Using Deep Learning) my approach involves implementing an ensemble of three deep learning models: ResNet50, MobileNetV2, and VGG16.
The plan is :

1.Train each model individually and evaluate their performance using metrics like accuracy, precision, and recall scores .
2.Combine these models' predictions using a weighted voting mechanism to improve classification accuracy.
3.Compare the ensemble’s results with individual models to demonstrate the advantages of using an ensemble approach.

@abhisheks008
Copy link
Owner

Assigned @KrishnaChaitanya16

@abhisheks008 abhisheks008 added Status: Assigned Assigned issue. and removed Status: Up for Grabs Up for grabs issue. labels Jan 26, 2025
@KrishnaChaitanya16
Copy link

@abhisheks008 Am I required to clone the whole repo or can I create a repo of my own and make that repo a submodule of the main repo

@abhisheks008
Copy link
Owner

Fork the repository and create a pull request to the main branch.

@KrishnaChaitanya16
Copy link

@abhisheks008 What are the accuracy scores expected for each model and the ensemble .

@abhisheks008
Copy link
Owner

Accuracy scores should be higher than 90%

@KrishnaChaitanya16
Copy link

KrishnaChaitanya16 commented Feb 12, 2025

@abhisheks008 I have changed my plan slightly , I have used ConvNxt instead of MobileNetV2 and i have achieved the following results

Classification Report for hair-convnext-tiny:

              precision    recall  f1-score   support

    Straight       0.92      0.96      0.94       195

        Wavy       0.87      0.82      0.84       126

       curly       0.89      0.95      0.92       186

  dreadlocks       0.98      0.98      0.98       193

       kinky       0.94      0.80      0.86        95

    accuracy                           0.92       795

   macro avg       0.92      0.90      0.91       795

weighted avg       0.92      0.92      0.92       795

=======================================

Classification Report for hair-vgg16:

              precision    recall  f1-score   support

    Straight       0.88      0.94      0.91       195

        Wavy       0.89      0.83      0.86       126

       curly       0.91      0.92      0.91       186

  dreadlocks       0.98      0.97      0.98       193

       kinky       0.90      0.85      0.88        95

    accuracy                           0.92       795

   macro avg       0.91      0.90      0.91       795

weighted avg       0.92      0.92      0.92       795

=========================================

Classification Report for hair-resnet50:

              precision    recall  f1-score   support

    Straight       0.92      0.92      0.92       195

        Wavy       0.82      0.81      0.82       126

       curly       0.86      0.92      0.89       186

  dreadlocks       0.97      0.99      0.98       193

       kinky       0.91      0.77      0.83        95

    accuracy                           0.90       795

   macro avg       0.90      0.88      0.89       795

weighted avg       0.90      0.90      0.90       795

============================================

Ensemble Model Classification Report:

              precision    recall  f1-score   support

    Straight       0.92      0.94      0.93       195

        Wavy       0.88      0.85      0.86       126

       curly       0.91      0.93      0.92       186

  dreadlocks       1.00      1.00      1.00       193

       kinky       0.90      0.86      0.88        95

    accuracy                           0.93       795

   macro avg       0.92      0.92      0.92       795

weighted avg       0.93      0.93      0.93       795

Is this ok? if yes i will proceed with the ensemble of these models.

@KrishnaChaitanya16 KrishnaChaitanya16 linked a pull request Feb 23, 2025 that will close this issue
12 tasks
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
Status: Assigned Assigned issue. WoC 4.0 Winter of Code 4.0 by GDG IIITK
Projects
None yet
Development

Successfully merging a pull request may close this issue.

4 participants