Xuezhi Zhang, Hongdong Zhao, Weina Liu, Yiming Zhao, Song Guan. An infrared vehicle detection method based on improved YOLOv5[J]. Infrared and Laser Engineering, 2023, 52(8): 20230245
- Infrared and Laser Engineering
- Vol. 52, Issue 8, 20230245 (2023)
Fig. 1. Improved YOLOv5 network architecture diagram
Fig. 2. Structure of CFG
Fig. 3. Structure of CA
Fig. 4. Structure of Spatial Attention
Fig. 5. (a) FPN structure, adds an upward path from small-sized feature map; (b) PANet structure, adds a downward path from large-sized feature map based on FPN; (c) BiFPN structure; (d) Z-BiFPN
Fig. 6. Diagram of Decoupled Head architecture
Fig. 7. (a) An example of a self-collected dataset; (b) An example of SCUT_FIR_Pedestrian_Dataset; (c) An example of MULTISPECTRAL DATASET
Fig. 8. PR curve. (a) YOLOv5; (b) Improved YOLOv5
Fig. 9. Confusion matrices. (a) YOLOv5; (b) Improved YOLOv5
Fig. 10. Comparison of detection results. (a) Original image; (b) YOLOv5; (c) Improved YOLOv5
|
Table 1. Experimental results of different improvement methods
|
Table 2. Comparison of different object detection algorithms
Set citation alerts for the article
Please enter your email address
© Copyright 2018-2021 | Chinese Laser Press. All Rights Reserved 沪ICP备15018463号-20