Skip to content

Commit df9d68e

Browse files
authored
Update README.md
1 parent 6fc3f5e commit df9d68e

File tree

1 file changed

+52
-8
lines changed
  • Hair Type Classification using Deep Learning v2/Model

1 file changed

+52
-8
lines changed
Original file line numberDiff line numberDiff line change
@@ -1,10 +1,54 @@
1-
## About the Dataset
2-
The Dataset had 5 classes of hair types that are curly, dreadlocks, kinky, Straight and Wavy
1+
## **PROJECT TITLE**
32

4-
## Models Implemented
5-
1. VGG16
6-
2. ResNet50V2
7-
3. InceptionResNetV2
3+
### 🎯 **Goal**
84

9-
## Performance Metrics
10-
Accuracy of 85%
5+
The goal of the project is to classify different hair types.
6+
7+
### 🧵 **Dataset**
8+
9+
The dataset is taken from Kaggle.
10+
Link: https://www.kaggle.com/datasets/kavyasreeb/hair-type-dataset
11+
12+
### 🧾 **Description**
13+
14+
The project is used to classify different hair types like staright, curly etc. It is useful to beauty parlours as they can recommend beauty products based on the hair type and also helpful for healthcare specialists as they can identify hair diseases for different hair types.
15+
16+
### 🧮 **What I had done!**
17+
18+
1. First I did some EDA on the data.
19+
2. Then I preprocessed the data and split it into training and validation sets.
20+
3. Then I trained 3 models on the data and also found out their performance metrics
21+
4. At the end I created an ensemble model with the 3 models which are ResNet50, VGG16 and InceptionResNet
22+
23+
### 🚀 **Models Implemented**
24+
25+
1. ResNet50 - I have chosen this because of its skip connections which solves the vanishing gradient problem allowing it to do better feature extraction.
26+
2. VGG16 - Chosen it as its simple yet effective architecture.
27+
3. InceptionResNet - Chosen for its combination of Inception modules and residual connections.
28+
29+
### 📚 **Libraries Needed**
30+
31+
- Matplotlib
32+
- pandas
33+
- numpy
34+
- sklearn
35+
- tensorflow (keras)
36+
37+
### 📊 **Exploratory Data Analysis Results**
38+
39+
![image](https://github.com/user-attachments/assets/f7de162c-66ac-4b7f-b490-44bd29b87d40)
40+
41+
42+
### 📈 **Performance of the Models based on the Accuracy Scores**
43+
44+
1. ResNet50 :- 85%
45+
2. VGG16 :- 84%
46+
3. InceptionResNet :- 84%
47+
48+
### 📢 **Conclusion**
49+
50+
ResNet50 performs the best for this project as it attains the highest accuracy of 85% out of the 3 models.
51+
52+
### ✒️ **Your Signature**
53+
54+
Subhrajyoti Basu

0 commit comments

Comments
 (0)