fahd9999
/

face_shape_classification

Image Classification

face shape classifier

Model card Files Files and versions

fahd9999 commited on Dec 30, 2024

Commit

9ffd76b

·

verified ·

1 Parent(s): 6ad7194

Update README.md

Files changed (1) hide show

README.md +14 -9

README.md CHANGED Viewed

@@ -1,12 +1,17 @@
----
-language:
-- en
-pipeline_tag: image-classification
-tags:
-- face
-- shape
-- face shape
----
 # Face Shape Classification Model (EfficientNetB4 + CNN)
 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.

+---
+pipeline_tag: image-classification
+tags:
+- face
+- shape
+- face shape
+- pytorch
+- torch
+- face model
+- face shape classifier
+datasets:
+- bkprocovid19/face_shape
+library_name: keras
+---
 # Face Shape Classification Model (EfficientNetB4 + CNN)
 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.