5 December 2023 Vehicle detection based on improved YOLOv5s using coordinate attention and decoupled head
Xuanjing Shen, Tongzhuang Liu, Yu Wang
Author Affiliations +
Abstract

Vehicle detection is a fundamental problem in object detection and plays a significant role in intelligent transportation and smart driving. To enhance the accuracy of vehicle detection and the robustness of the model in detecting occluded vehicles, we propose an improved vehicle detection method for you only look once v5s (YOLOv5s). First, we introduce the coordinate attention module into the backbone of the model. This module guides the model to improve its attention toward the location information of vehicles and channel features under occlusion conditions. Second, the feature fusion component of the model is improved by incorporating bidirectional scale connections and weighted feature fusion. Finally, the prediction head of YOLOv5s is decoupled and the regression and classification tasks are assigned to two separate branches. Experimental results show that our proposed method has 2% and 2.5% higher average precision than YOLOv5s for the common objects in context vehicle dataset and University at Albany Detection and Tracking, respectively.

© 2023 SPIE and IS&T
Xuanjing Shen, Tongzhuang Liu, and Yu Wang "Vehicle detection based on improved YOLOv5s using coordinate attention and decoupled head," Journal of Electronic Imaging 32(6), 063023 (5 December 2023). https://doi.org/10.1117/1.JEI.32.6.063023
Received: 23 July 2023; Accepted: 31 October 2023; Published: 5 December 2023
Advertisement
Advertisement
RIGHTS & PERMISSIONS
Get copyright permission  Get copyright permission on Copyright Marketplace
KEYWORDS
Object detection

Head

Feature fusion

Detection and tracking algorithms

Education and training

Feature extraction

Target detection

Back to Top