Laser & Optoelectronics Progress, Volume. 59, Issue 18, 1815016(2022)
Small Target Accurate Vehicle Detection Algorithm Based on Improved Transformer
Figures & Tables(10)
Fig. 5. Swin Transformer block and SW-MSA. (a) Swin Transformer block; (b) W-MSA; (c) SW-MSA
Fig. 8. Training process of Swin Transformer and proposed model. (a) Swin Transformer; (b) proposed model
Fig. 9. Contrast of sharpness and detection effect between original image and image enhancement. (a) (b) (c) (d) Illumination change; (e) (f) (g) (h) blurred, dense small targets; (i) (j) (k) (l) long range small targets
Table 1. Comparison results of different detection algorithm performance
View tableTable 1. Comparison results of different detection algorithm performance
Testset Model Backbone AP /% Time /ms UA-DETRAC Cascade R-CNN ResNet-101 79.8 48.8 UA-DETRAC YOLOV3 Darknet-53 87.6 42.3 UA-DETRAC Swin-Transformer Swin-T 96.4 65.3 UA-DETRAC Proposed model Enhance-Swin-T 99.0 68.4 VOC-vehicle Cascade R-CNN ResNet-101 86.7 48.8 VOC-vehicle YOLOV3 Darknet-53 90.9 42.3 VOC-vehicle Swin-Transformer Swin-T 95.2 65.3 VOC-vehicle Proposed model Enhance-Swin-T 98.1 68.4 Small-100 Cascade R-CNN ResNet-101 49.3 48.8 Small-100 YOLOV3 Darknet-53 57.9 42.3 Small-100 Swin-Transformer Swin-T 80.4 65.3 Small-100 Proposed model Enhance-Swin-T 88.3 68.4
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
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)
© Copyright 2018-2021 | Chinese Laser Press.
All Rights Reserved 沪ICP备15018463号-20