Abstract
In the recognition of distracted driving behaviour, traditional manual feature extraction is subjective and complex; single deep convolutional network also has problems such as insufficient generalisation performance and stability. To solve the above problems, this paper proposes a distracted driving behaviour recognition method based on transfer learning and model fusion. First, based on the transfer learning method, the deep convolutional neural network models ResNet18 and ResNet34 are used to extract the features of some images, respectively. Furthermore, the pre-trained model is fine-tuned to obtain four deep convolutional neural network models. Finally, the four network models are fused by stacking method, using 5-fold cross-validation method to reduce over-fitting. Experimental results show that the recognition accuracy of distracted driving behaviour after model fusion reaches 95.47%. The fusion model has higher model generalisation performance and recognition accuracy, which can provide certain technical support for the research of distracted driving behaviour recognition.
Published Version
Talk to us
Join us for a 30 min session where you can share your feedback and ask us any queries you have
Schedule a call