An Algorithm for Distracted Driving Recognition Based on Pose Features and an Improved KNN

Yingjie Gong,Xizhong Shen

DOI: https://doi.org/10.3390/electronics13091622

IF: 2.9

2024-04-25

Electronics

Abstract:To reduce safety accidents caused by distracted driving and address issues such as low recognition accuracy and deployment difficulties in current algorithms for distracted behavior detection, this paper proposes an algorithm that utilizes an improved KNN for classifying driver posture features to predict distracted driving behavior. Firstly, the number of channels in the Lightweight OpenPose network is pruned to predict and output the coordinates of key points in the upper body of the driver. Secondly, based on the principles of ergonomics, driving behavior features are modeled, and a set of five-dimensional feature values are obtained through geometric calculations. Finally, considering the relationship between the distance between samples and the number of samples, this paper proposes an adjustable distance-weighted KNN algorithm (ADW-KNN), which is used for classification and prediction. The experimental results show that the proposed algorithm achieved a recognition rate of 94.04% for distracted driving behavior on the public dataset SFD3, with a speed of up to 50FPS, superior to mainstream deep learning algorithms in terms of accuracy and speed. The superiority of ADW-KNN was further verified through experiments on other public datasets.

engineering, electrical & electronic,computer science, information systems,physics, applied

What problem does this paper attempt to address?

### What problem does this paper attempt to solve?

This paper aims to address several key issues in distracted driving recognition:

1. **Low recognition accuracy**: Current distracted driving detection algorithms suffer from low recognition accuracy.
2. **Deployment difficulties**: Existing algorithms face certain challenges when deployed in real-world scenarios.

To tackle these problems, the authors propose a method based on an improved KNN (ADW-KNN) algorithm to identify distracted driving behaviors. The specific steps are as follows:

1. **Key point extraction**: Use a lightweight OpenPose network to extract the key point coordinates of the driver's upper body.
2. **Feature modeling**: Establish a driving behavior feature model based on ergonomic principles and obtain five-dimensional feature values through geometric calculations.
3. **Classification prediction**: Propose an adjustable distance-weighted KNN (ADW-KNN) algorithm for classifying and predicting distracted driving behaviors.

Experimental results show that this algorithm achieves a distracted driving recognition rate of 94.04% on the public dataset SFD3, with a processing speed of up to 50FPS, outperforming mainstream deep learning algorithms. Additionally, experiments on other public datasets further validate the advantages of the ADW-KNN algorithm.