Hugging Face's logo

fahd9999
/
face_shape_classification

Image Classification
Keras
PyTorch
face
shape
face shape
torch
face model
face shape classifier
Model card Files Community
fahd9999 commited on
Commit
9ffd76b
·
verified ·
1 Parent(s): 6ad7194

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +14 -9
README.md CHANGED
@@ -1,12 +1,17 @@
1
- ---
2
- language:
3
- - en
4
- pipeline_tag: image-classification
5
- tags:
6
- - face
7
- - shape
8
- - face shape
9
- ---
 
 
 
 
 
10
  # Face Shape Classification Model (EfficientNetB4 + CNN)
11
 
12
  This model is designed for **face shape classification**. It uses **EfficientNetB4** as the backbone, trained with transfer learning, and fine-tuned with a custom Convolutional Neural Network (CNN) to classify images into 5 different face shape categories. The model uses a **softmax activation function** to return the probability distribution over the 5 classes.
 
1
+ ---
2
+ pipeline_tag: image-classification
3
+ tags:
4
+ - face
5
+ - shape
6
+ - face shape
7
+ - pytorch
8
+ - torch
9
+ - face model
10
+ - face shape classifier
11
+ datasets:
12
+ - bkprocovid19/face_shape
13
+ library_name: keras
14
+ ---
15
  # Face Shape Classification Model (EfficientNetB4 + CNN)
16
 
17
  This model is designed for **face shape classification**. It uses **EfficientNetB4** as the backbone, trained with transfer learning, and fine-tuned with a custom Convolutional Neural Network (CNN) to classify images into 5 different face shape categories. The model uses a **softmax activation function** to return the probability distribution over the 5 classes.