HIC-YOLOv5：改进的YOLOv5，用于小物体检测

摘要

论文链接：https://arxiv.org/pdf/2309.16393.pdf
小目标检测一直是目标检测领域的一个具有挑战性的问题。有一些工作提出了针对此任务的改进，例如添加多个注意力块或更改特征融合网络的整个结构。然而，这些模型的计算成本很大，这使得部署实时目标检测系统变得不可行，同时留下了改进的空间。为此，提出了一个改进的YOLOv5模型：HICYOLOv5来解决上述问题。首先，添加了一个特定于小目标的额外预测头，以提供更高分辨率的特征图以进行更好的预测。其次，在主干和颈部之间采用了一个内卷积块，以增加特征图的通道信息。此外，在主干的末端应用了一种名为CBAM的注意力机制，从而不仅降低了与以前工作相比的计算成本，而且强调了通道和空间域中的重要信息。我们的结果表明，HIC-YOLOv5在VisDrone-2019-DET数据集上的mAP@[.5:.95]提高了6.42%，[email protected]提高了9.38%。

1、简介

目标检测算法已广泛应用于无人机的智能系统，如行人检测和车辆检测。它使无人机拍摄的照片的分析过程自动化。然而，这类应用的最大问题在于检测小物体，因为从更高的高度拍摄的照片中的大多数物体都会变小。这一事实对