22 December 2023 UTD-YOLO: underwater trash detection model based on improved YOLOv5
Guanghao Wu, Yan Ge, Qian Yang
Author Affiliations +
Abstract

Underwater trash detection faces many problems. For example, the resolution of underwater trash datasets is generally low, and trash may show irregular shapes, sizes, and scales, especially small objects that are difficult to detect. To solve the above problems, we propose UTD-YOLO, an improved YOLOv5s trash detection model. This model proposes a cross-layer aggregation spatial dimensionality reduction module (CASDRM) and channel-PAN-FPN with improved channel information. CASDRM preserves the information of the channel dimension and utilizes the receptive field between different layers to enhance the feature extraction ability while strengthening the ability to adapt to the geometric changes of objects. Channel-PAN-FPN realizes two-way information transmission and improves the recognition accuracy of the model. UTD-YOLO can achieve good recognition results for smaller-scale targets in low-resolution datasets. The experimental results show that the UTD-YOLO model improves the mAP50 to 73.2% on the Trashcan dataset, which is higher than the original YOLOv5 algorithm.

© 2023 SPIE and IS&T
Guanghao Wu, Yan Ge, and Qian Yang "UTD-YOLO: underwater trash detection model based on improved YOLOv5," Journal of Electronic Imaging 32(6), 063034 (22 December 2023). https://doi.org/10.1117/1.JEI.32.6.063034
Received: 27 July 2023; Accepted: 1 December 2023; Published: 22 December 2023
Advertisement
Advertisement
RIGHTS & PERMISSIONS
Get copyright permission  Get copyright permission on Copyright Marketplace
KEYWORDS
Submerged target modeling

Object detection

Target detection

Feature extraction

Deformation

Detection and tracking algorithms

Convolution

Back to Top