YOLOv7 Vehicle Target Detection Algorithm Based on Attention Mechanism
While pedestrian target detection methods have advanced considerably, vehicle target detection still holds significant potential in the field of intelligent identification. However, this task faces challenges due to factors like lighting, environmental conditions, and occlusion, which could affect detection accuracy. This study focuses on vehicle target detection, aiming to address the difficulties of detecting and avoiding missed detections of vehicles in complex backgrounds. By utilizing the YOLOv7 algorithm, we constructed a dataset that includes complex backgrounds and small-scale vehicle images, integrated the Coordinate Attention mechanism, adding CA attention mechanism, designing the algorithm model YOLOv7-veh, training the model, and performing target recognition on vehicles respectively, the superiority of YOLOv7-veh algorithm in complex background and small-scale vehicle target detection is proved through comparative tests.
Authors
- Lidan Li (ORCID: https://orcid.org/0000-0001-7551-4469)
- Gaotian Liu
- Xiaozheng Zhang (ORCID: https://orcid.org/0000-0003-2449-057X)
- Chenyu Gu
- Hong Du
- Zhonglin Yang
Institutions
- State Key Laboratory of Vehicle NVH and Safety Technology (CN)
Publication Details
- Published
- 2024-10-23
- DOI
- https://doi.org/10.1109/iccasit62299.2024.10828068
- Citations
- 2
- Primary Topic
- Industrial Vision Systems and Defect Detection
- Type
- article
- Field-Weighted Citation Impact
- 0.70