Model Description

This model is a fine-tuned version of google/vit-base-patch16-224-in21k. It has been fine-tuned on the FER2013 dataset, which contains facial expression images. The model classifies images into four categories of facial expressions:

Recommendations

Users (both direct and downstream) should be made aware of the risks, biases and limitations of the model. More information needed for further recommendations.

How to Get Started with the Model

Use the code below to get started with the model.

[More Information Needed]

Evaluation

Results

Results Summary Training Accuracy: 83.27% Validation Accuracy: 76.52%

Downloads last month
5
Safetensors
Model size
85.8M params
Tensor type
F32
·
Inference Examples
Unable to determine this model's library. Check the docs .

Model tree for gerhardien/face-emotion

Finetuned
(1769)
this model

Dataset used to train gerhardien/face-emotion