本文主要是介绍HIC-YOLOv5:改进的YOLOv5,用于小物体检测,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!
文章目录
- 摘要
- 1、简介
- 2、相关工作
- 3、方法
- 4、实验结果
- 5、结论
摘要
论文链接:https://arxiv.org/pdf/2309.16393.pdf
小目标检测一直是目标检测领域的一个具有挑战性的问题。有一些工作提出了针对此任务的改进,例如添加多个注意力块或更改特征融合网络的整个结构。然而,这些模型的计算成本很大,这使得部署实时目标检测系统变得不可行,同时留下了改进的空间。为此,提出了一个改进的YOLOv5模型:HICYOLOv5来解决上述问题。首先,添加了一个特定于小目标的额外预测头,以提供更高分辨率的特征图以进行更好的预测。其次,在主干和颈部之间采用了一个内卷积块,以增加特征图的通道信息。此外,在主干的末端应用了一种名为CBAM的注意力机制,从而不仅降低了与以前工作相比的计算成本,而且强调了通道和空间域中的重要信息。我们的结果表明,HIC-YOLOv5在VisDrone-2019-DET数据集上的mAP@[.5:.95]提高了6.42%,mAP@0.5提高了9.38%。
1、简介
目标检测算法已广泛应用于无人机的智能系统,如行人检测和车辆检测。它使无人机拍摄的照片的分析过程自动化。然而,这类应用的最大问题在于检测小物体,因为从更高的高度拍摄的照片中的大多数物体都会变小。这一事实对
这篇关于HIC-YOLOv5:改进的YOLOv5,用于小物体检测的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!