Distracted-Driver-Detection icon indicating copy to clipboard operation
Distracted-Driver-Detection copied to clipboard

CNN model that detects distracted drivers on the road

Distracted-Driver-Detection

CNN model that detects distracted drivers on the road

This is a comptetition that I saw in Kaggle: "State Farm Distracted Driver Detection". You cand find more information here: https://www.kaggle.com/c/state-farm-distracted-driver-detection

I decided to use the same dataset and create a CNN model in PyTorch to detect 10 different types of distracted drivers. The 10 categories are:

-c0: safe driving -c1: texting - right -c2: talking on the phone - right -c3: texting - left -c4: talking on the phone - left -c5: operating the radio -c6: drinking -c7: reaching behind -c8: hair and makeup -c9: talking to passenger

If you want to see the results from this experiement, check out the notebook: https://github.com/viritaromero/Distracted-Driver-Detection/blob/master/Detecting_Driver_Distraction_pynb.ipynb