|
--- |
|
base_model: |
|
- openai/clip-vit-base-patch16 |
|
--- |
|
|
|
# Fine-Tuned ResNet-50 on FERPlus Dataset |
|
|
|
This model is a fine-tuned version of ViT on the [FERPlus dataset](https://www.kaggle.com/datasets/arnabkumarroy02/ferplus), which is more balanced ferplus dataset as owner claimed. |
|
|
|
## Model Details |
|
- **Base Model**: [Clip-ViT-Base-Patch16](https://huggingface.co/openai/clip-vit-base-patch16) |
|
- **Dataset**: FERPlus, which contains grayscale images of faces labeled with emotion categories. |
|
- **Task**: Emotion Classification |
|
- **Accruacy** %82.4 |
|
- **Labels**: |
|
- 0: Angry |
|
- 1: Contempt |
|
- 2: Disgust |
|
- 3: Fear |
|
- 4: Happy |
|
- 5: Neutral |
|
- 6: Sad |
|
- 7: Surprise |
|
|
|
### Training Hyperparameters |
|
- **Batch Size**: 16 |
|
- **Epochs**: 1 |
|
- **Learning Rate**: 2e-5 |
|
- **Weight Decay**: 0.01 |
