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