Opto-Electronic Engineering, Volume. 48, Issue 4, 200325(2021)
An end-to-end neural network for mobile phone detection in driving scenarios
Real-time detection of small objects is always a difficult problem in image processing. Based on the target detection algorithm of deep learning, this paper proposed an end-to-end neural network for mobile phone small target detection in complex driving scenarios. Firstly, an end-to-end small target detection network (OMPDNet) was designed to extract image features by improving the YOLOv4 algorithm. Secondly, based on the K-means algorithm, a K-means-Precise clustering algorithm of more appropriate data samples distribution in the clustering center was designed, which was used to generate prior frames suitable for small target data, so as to improve the efficiency of the network model. Finally, we constructed our own data set with supervision and weak supervision, and added negative samples to the data set for training. In the complex driving scene experiments, the OMPDNet algorithm proposed in this paper can not only effectively complete the detection task of using mobile phone while driving, but also has certain advantages over the current popular algorithms in accuracy and real-time for small target detection.
Get Citation
Copy Citation Text
Dai Teng, Zhang Ke, Yin Dong. An end-to-end neural network for mobile phone detection in driving scenarios[J]. Opto-Electronic Engineering, 2021, 48(4): 200325
Category: Article
Received: Sep. 2, 2020
Accepted: --
Published Online: Sep. 4, 2021
The Author Email: Dong Yin (yindong@ustc.edu.cn)