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

• Angry
• Disgust
• Fear
• Happy
• Sad
• Surprise
• Neutral

Data Preprocessing

• 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.