Sep, 2023

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

TL;DR在小物体检测领域,针对计算代价大且预测性能有待提高的问题,提出了一种改进的 YOLOv5 模型:HIC-YOLOv5。它通过添加特定于小物体的额外预测头部以提供更高分辨率的特征图进行更精确的预测,采用了 involution 块在主干网和 neck 之间以增加特征图的通道信息,并在主干网的末端应用了一种名为 CBAM 的注意力机制,从而不仅减少了与先前方法相比的计算代价,还加强了通道和空间域中的重要信息。在 VisDrone-2019-DET 数据集上,HIC-YOLOv5 的 mAP@[.5:.95] 提高了 6.42%,[email protected] 提高了 9.38%。