# Load model directly
from transformers import AutoImageProcessor, AutoModelForImageClassification
processor = AutoImageProcessor.from_pretrained("trpakov/vit-face-expression")
model = AutoModelForImageClassification.from_pretrained("trpakov/vit-face-expression")Quick Links
Vision Transformer (ViT) for Facial Expression Recognition Model Card
Model Overview
Model Name: trpakov/vit-face-expression
Task: Facial Expression/Emotion Recognition
Dataset: FER2013
Model Architecture: Vision Transformer (ViT)
Finetuned from model: vit-base-patch16-224-in21k
Model Description
The vit-face-expression model is a Vision Transformer fine-tuned for the task of facial emotion recognition.
It is trained on the FER2013 dataset, which consists of facial images categorized into seven different emotions:
- Angry
- Disgust
- Fear
- Happy
- Sad
- Surprise
- Neutral
Data Preprocessing
The input images are preprocessed before being fed into the model. The preprocessing steps include:
- Resizing: Images are resized to the specified input size.
- Normalization: Pixel values are normalized to a specific range.
- Data Augmentation: Random transformations such as rotations, flips, and zooms are applied to augment the training dataset.
Evaluation Metrics
- Validation set accuracy: 0.7113
- Test set accuracy: 0.7116
Limitations
- Data Bias: The model's performance may be influenced by biases present in the training data.
- Generalization: The model's ability to generalize to unseen data is subject to the diversity of the training dataset.
- Downloads last month
- 14,462
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("image-classification", model="trpakov/vit-face-expression") pipe("https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/hub/parrots.png")