StephanieNhiLe / drowsy-guard

1 stars 0 forks source link

Train a model with good prediction #3

Closed PV1shal closed 2 months ago

PV1shal commented 2 months ago

Train model on dataset: https://www.kaggle.com/datasets/ismailnasri20/driver-drowsiness-dataset-ddd Compare with hugging face model: https://huggingface.co/chbh7051/vit-driver-drowsiness-detection

Finalize what model to use 👍