计算机科学
水下
频道(广播)
人工智能
特征提取
计算机视觉
降维
模式识别(心理学)
维数(图论)
维数之咒
特征(语言学)
图像分辨率
追踪
空间分析
遥感
数学
电信
语言学
海洋学
哲学
纯数学
地质学
操作系统
作者
Guangning Wu,Yan Ge,Qiong Yang
标识
DOI:10.1117/1.jei.32.6.063034
摘要
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.
科研通智能强力驱动
Strongly Powered by AbleSci AI