| --- |
| language: en |
| license: mit |
| tags: |
| - image-classification |
| - face-shape |
| - tensorflow |
| - mobilenetv2 |
| base_model: google/mobilenet_v2 |
| datasets: |
| - niten19/face-shape-dataset |
| metrics: |
| - accuracy |
| model-index: |
| - name: face-shape-classifier |
| results: |
| - task: |
| type: image-classification |
| dataset: |
| name: face-shape-dataset |
| type: niten19/face-shape-dataset |
| metrics: |
| - type: accuracy |
| value: 0.63 |
| --- |
| # Face Shape Classifier |
|
|
| ## Project Overview |
| AI-powered app that detects face shape from photo and suggests best hairstyle. |
|
|
| ## Model Details |
| - Architecture: MobileNetV2 (Transfer Learning) |
| - Framework: TensorFlow/Keras |
| - Validation Accuracy: 63% |
| - Epochs: 25 |
|
|
| ## Dataset |
| - Source: Kaggle - niten19/face-shape-dataset |
| - Classes: Heart, Oval, Round, Square, Oblong |
| - Training Images: 4000 |
| - Testing Images: 1000 |
|
|
| ## Tech Stack |
| - Python |
| - TensorFlow |
| - MediaPipe |
| - OpenCV |
| - Streamlit |
|
|
| ## ML Pipeline |
| 1. Photo Upload |
| 2. Face Detection (MediaPipe) |
| 3. Face Shape Classification (MobileNetV2) |
| 4. Hairstyle Recommendation |
| 5. Virtual Try-On Overlay |
