DistractedDriverClassification
Introduction.
The following is a model that I trained on distracted driver dataset. Feel free to fork and contribute.
Citation.
Y. Abouelnaga, H. Eraqi, and M. Moustafa. "Real-time Distracted Driver Posture Classification". Neural Information Processing Systems (NIPS 2018), Workshop on Machine Learning for Intelligent Transportation Systems, Dec. 2018.
Requirements.
Pytorch is required. I have tested this on Python 3.7 on Ubuntu 18.2 LTS.
pip install pytorch torchvision
Usage.
python src/main.py