Downloads · 30 days
4
9% of all-time downloads
jeremoo/vit-fef-finetuned
vit-fef-finetuned is a machine learning model from jeremoo. Use it for the machine learning task on the model card, and read the license before you ship it in a product.
Downloads · 30 days
4
9% of all-time downloads
All-time downloads
46
Public
Parameters
85.8M
343 MB on disk
Likes
0
Public
Click a slice to open those files.
.safetensors343 MB · 100%
From the Hugging Face model README
For EmoryHacks:
I used an NVIDIA H100 through a DigitalOcean GPU Droplet to finetune Google’s Vision Transformer (ViT) model pre-trained on ImageNet-21k using a Facial Expression of Fatigues (FEF) dataset from Kaggle.
Base Model: https://huggingface.co/google/vit-base-patch16-224
Dataset: https://www.kaggle.com/datasets/ismailnasri20/driver-drowsiness-dataset-ddd