Distracted-Driver-Detection
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