返回
MC-YOLOv5: A Multi-Class Small Object Detection Algorithm
DOI:10.3390/biomimetics8040342.png)
摘要
En 中文
The detection of multi-class small objects poses a significant challenge in the field of computer vision. While the original YOLOv5 algorithm is more suited for detecting full-scale objects, it may not perform optimally for this specific task. To address this issue, we proposed MC-YOLOv5, an algorithm specifically designed for multi-class small object detection. Our approach incorporates three key innovations: (1) the application of an improved CB module during feature extraction to capture edge information that may be less apparent in small objects, thereby enhancing detection precision; (2) the introduction of a new shallow network optimization strategy (SNO) to expand the receptive field of convolutional layers and reduce missed detections in dense small object scenarios; and (3) the utilization of an anchor frame-based decoupled head to expedite training and improve overall efficiency. Extensive evaluations on VisDrone2019, Tinyperson, and RSOD datasets demonstrate the feasibility of MC-YOLOv5 in detecting multi-class small objects. Taking VisDrone2019 dataset as an example, our algorithm outperforms the original YOLOv5L with improvements observed across various metrics: mAP50 increased by 8.2%, mAP50-95 improved by 5.3%, F1 score increased by 7%, inference time accelerated by 1.8 ms, and computational requirements reduced by 35.3%. Similar performance gains were also achieved on other datasets. Overall, our findings validate MC-YOLOv5 as a viable solution for accurate multi-class small object detection.
Keyword:
YOLOv5
multi-class
small objects
shallow network optimization
CB structure
AI总结
对已上传原文的论文进行重点信息的提取,主要内容包括:简要概述、研究摘要、背景介绍、关键亮点、图文解析、展望与总结。
期刊
B
IF:
3.9
论文数:
3.2K
被引数:
5.1K
机构
引用论文
Small Object Detection in Traffic Scenes Based on Attention Feature Fusion基于注意力特征融合的交通场景小目标检测
SENSORS
IF3.5
The Use of Ranks to Avoid the Assumption of Normality Implicit in the Analysis of Variance使用秩来避免方差分析中隐含的正态性假设
Accurate Object Localization in Remote Sensing Images Based on Convolutional Neural Networks基于卷积神经网络的遥感图像目标精确定位
Elliptic Fourier transformation-based histograms of oriented gradients for rotationally invariant object detection in remote-sensing images基于椭圆傅里叶变换的方向梯度直方图,用于遥感图像中的旋转不变目标检测
Developing Wind and/or Solar Powered Crop Irrigation Systems for the Great Plains为大平原开发风能和/或太阳能作物灌溉系统

