Laser & Optoelectronics Progress, Volume. 59, Issue 18, 1815016(2022)
Small Target Accurate Vehicle Detection Algorithm Based on Improved Transformer
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%.
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
Category: Machine Vision
Received: Jun. 11, 2021
Accepted: Sep. 2, 2021
Published Online: Aug. 30, 2022
The Author Email: Li Yang (2442590777@qq.com)