This paper evaluates the performance of YOLOv5 in detecting small targets and proposes structural improvements to enhance its capabilities. Firstly, an additional small targets detection head is integrated into the original three-head architecture. To enrich feature map channel information, the convolutional block attention module (CBAM) is embedded in the backbone, while the involution attention mechanism is incorporated into the neck. Furthermore, the original intersection over union (IoU) loss is combined with the normalized wasserstein distance (NWD) loss to reduce sensitivity to positional deviations in small targets. These modifications improve the ability of the model together to detect and localize small targets effectively.

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

An Enhanced YOLOv5 Method via Hybrid NWD Loss and Dual Attention for Small Target Detection

  • Liu Silin

摘要

This paper evaluates the performance of YOLOv5 in detecting small targets and proposes structural improvements to enhance its capabilities. Firstly, an additional small targets detection head is integrated into the original three-head architecture. To enrich feature map channel information, the convolutional block attention module (CBAM) is embedded in the backbone, while the involution attention mechanism is incorporated into the neck. Furthermore, the original intersection over union (IoU) loss is combined with the normalized wasserstein distance (NWD) loss to reduce sensitivity to positional deviations in small targets. These modifications improve the ability of the model together to detect and localize small targets effectively.