Optimally-Weighted Image-Pose Approach (OWIPA) for Distracted Driver Detection and Classification.

Journal: Sensors (Basel, Switzerland)
PMID:

Abstract

Distracted driving is the prime factor of motor vehicle accidents. Current studies on distraction detection focus on improving distraction detection performance through various techniques, including convolutional neural networks (CNNs) and recurrent neural networks (RNNs). However, the research on detection of distracted drivers through pose estimation is scarce. This work introduces an ensemble of ResNets, which is named Optimally-weighted Image-Pose Approach (OWIPA), to classify the distraction through original and pose estimation images. The pose estimation images are generated from HRNet and ResNet. We use ResNet101 and ResNet50 to classify the original images and the pose estimation images, respectively. An optimum weight is determined through grid search method, and the predictions from both models are weighted through this parameter. The experimental results show that our proposed approach achieves 94.28% accuracy on AUC Distracted Driver Dataset.

Authors

  • Hong Vin Koay
    Department of Electrical Engineering, Faculty of Engineering, University of Malaya, Kuala Lumpur 50603, Malaysia.
  • Joon Huang Chuah
    Department of Electrical Engineering, Faculty of Engineering, University of Malaya, 40603 Kuala Lumpur, Malaysia.
  • Chee-Onn Chow
    Department of Electrical Engineering, Faculty of Engineering, University of Malaya, Kuala Lumpur 50603, Malaysia.
  • Yang-Lang Chang
    Department of Electrical Engineering, National Taipei University of Technology, Taipei 10608, Taiwan.
  • Bhuvendhraa Rudrusamy
    School of Engineering and Physical Sciences, Heriot-Watt University Malaysia, Putrajaya 62200, Malaysia.