|
--- |
|
license: mit |
|
base_model: |
|
- google/vit-base-patch16-224-in21k |
|
pipeline_tag: image-classification |
|
datasets: |
|
- Jeneral/fer-2013 |
|
--- |
|
|
|
### 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 |
|
|
|
<!-- This section is meant to convey recommendations with respect to the bias, risk, and technical limitations. --> |
|
|
|
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 |
|
|
|
<!-- This section describes the evaluation protocols and provides the results. --> |
|
|
|
### Results |
|
|
|
Results Summary |
|
Training Accuracy: 83.27% |
|
Validation Accuracy: 76.52% |
