Laser & Optoelectronics Progress, Volume. 59, Issue 18, 1815016(2022)

Small Target Accurate Vehicle Detection Algorithm Based on Improved Transformer

Guangda Xie1,2, Yang Li2、*, Hongquan Qu2, and Zaiming Sun3
Author Affiliations
  • 1School of Electrical and Control Engineering, North China University of Technology, Beijing 100144, China
  • 2Information College, North China University of Technology, Beijing 100144, China
  • 3School of Control and Computer Engineering, North China Electric Power University, Beijing 102206, China
  • show less

    Intelligent transportation systems have been playing a major role in vehicle detection technology. Recently, the convolutional neural network (CNN) architecture is a popular method for vehicle detection. However, in complex traffic situations, only fewer pixels for long-distance small targets are available, and CNN’s subsampling mechanism seems to be lacking sufficient context information of some extracted features, which gives small target detection great challenges. A small target vehicle detection algorithm based on a visual Transformer was introduced in this paper to solve the aforesaid problem. By improving the linear embedding module of the Transformer, information on the small targets was supplemented. Additionally, the image was constructed hierarchically, and each layer was only related to the part. Modeling, while expanding, the receptive field, instead of CNN, was conducted to extract more powerful features from small target vehicles to achieve accurate end-to-end detection. Data was verified using the UA-DETRAC vehicle dataset. The experimental results showed that the improved vehicle detection algorithm enhanced the detection performance of small targets at long distances and under severe occlusion conditions and that the detection accuracy reached 99.0%.

    Tools

    Get Citation

    Copy Citation Text

    Guangda Xie, Yang Li, Hongquan Qu, Zaiming Sun. Small Target Accurate Vehicle Detection Algorithm Based on Improved Transformer[J]. Laser & Optoelectronics Progress, 2022, 59(18): 1815016

    Download Citation

    EndNote(RIS)BibTexPlain Text
    Save article for my favorites
    Paper Information

    Category: Machine Vision

    Received: Jun. 11, 2021

    Accepted: Sep. 2, 2021

    Published Online: Aug. 30, 2022

    The Author Email: Li Yang (2442590777@qq.com)

    DOI:10.3788/LOP202259.1815016

    Topics