|
1 |
| -## About the Dataset |
2 |
| -The Dataset had 5 classes of hair types that are curly, dreadlocks, kinky, Straight and Wavy |
| 1 | +## **PROJECT TITLE** |
3 | 2 |
|
4 |
| -## Models Implemented |
5 |
| -1. VGG16 |
6 |
| -2. ResNet50V2 |
7 |
| -3. InceptionResNetV2 |
| 3 | +### 🎯 **Goal** |
8 | 4 |
|
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 | + |
| 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