返回
Improved Small Object Detection Algorithm Based on YOLOv5
DOI:10.1109/MIS.2024.3399053.png)
摘要
En 中文
YOLOv5 is a popular object detection algorithm that is widely used in various industrial fields, especially in the field of autonomous driving. However, this algorithm has problems, such as false positives and false negatives when detecting small targets. The article proposes an improved method for small object detection using YOLOv5s. First, a multilevel feature fusion detection head is proposed to extract larger feature maps from the backbone of the model, improving the ability to extract features of small objects. Second, a decoupled attention mechanism is introduced at each detection head, which separates the detection of object box position, object box confidence, and class probability to reduce confusion between different feature information. Finally, the focal minimum points distance intersection over union loss function is adopted to mitigate the effects of class imbalance and poor-quality object pixels.
Keyword:
Feature extraction
YOLO
Head
Semantics
Intelligent systems
Neck
Remote sensing
期刊
IF:
6.1
论文数:
1.6K
被引数:
4.5K
机构
引用论文
Recycling of NdFeB Magnets from Electric Drive Motors of (Hybrid) Electric Vehicles从 (混合动力) 电动汽车的驱动电机中回收NdFeB磁体

