• Opto-Electronic Engineering
  • Vol. 48, Issue 4, 200325 (2021)
Dai Teng1、2, Zhang Ke1、2, and Yin Dong1、2、*
Author Affiliations
  • 1[in Chinese]
  • 2[in Chinese]
  • show less
    DOI: 10.12086/oee.2021.200325 Cite this Article
    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 Copy Citation Text show less

    Abstract

    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.
    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
    Download Citation