14 July 2023 DBCR-YOLO: improved YOLOv5 based on double-sampling and broad-feature coordinate-attention residual module for water surface object detection
Yanyu Guo, Xiaomei Tian, Yanting Xiao
Author Affiliations +
Abstract

Unmanned missions have become more and more popular in recent years. The related technologies of unmanned ground vehicles and unmanned aerial vehicles are growing rapidly, but research on unmanned surface vehicles (USVs) is rare. Water surface object detection algorithms play a crucial role in the field of USVs. However, achieving an object detection algorithm that balances speed and accuracy in the presence of interference is a difficult challenge. We proposed a network, DBCR-YOLO, that improved the detection accuracy while meeting real-time requirements. Based on YOLOv5, we added an additional detection head for detecting tiny objects. Then, we replaced the downsampling in YOLOv5’s backbone network with the proposed double sampling mechanism to solve the problem that paying attention to the key features of objects cannot be done in the downsampling process of YOLOv5. Finally, we substituted the proposed BCR neck for YOLOv5’s neck, thus improving the fusion of features between different scales based on fewer parameters and fewer calculations. We tested our network on the water surface object detection dataset. Compared with YOLOv5, DBCR-YOLO improved the detection accuracy by 3.4%. At the same time, DBCR-YOLO achieved the highest accuracy in comparison with other networks.

© 2023 SPIE and IS&T
Yanyu Guo, Xiaomei Tian, and Yanting Xiao "DBCR-YOLO: improved YOLOv5 based on double-sampling and broad-feature coordinate-attention residual module for water surface object detection," Journal of Electronic Imaging 32(4), 043013 (14 July 2023). https://doi.org/10.1117/1.JEI.32.4.043013
Received: 22 February 2023; Accepted: 28 June 2023; Published: 14 July 2023
Lens.org Logo
CITATIONS
Cited by 1 scholarly publication.
Advertisement
Advertisement
RIGHTS & PERMISSIONS
Get copyright permission  Get copyright permission on Copyright Marketplace
KEYWORDS
Object detection

Detection and tracking algorithms

Neck

Head

Convolution

Feature fusion

Feature extraction

Back to Top