Abstract:
提出一种多层特征图信息融合的海滩小目标检测方法,从上下文信息与强化特征图信息融合的角度提升小目标游客的检出率.首先,透过更全面、有效的GAM注意力机制思想结合CSP结构提出GCSAM结构,用于增强检测YOLOv5模型中主干网络跨纬度感受区,聚焦小目标特征学习;其次,在颈部融合端使用BIFPN结构优化YOLOv5网络中PANet结构,补全跨层特征信息之间的传递,使得特征图包含更多的上下文信息;最后,采用幂变换改进YOLOv5网络中CI-OU_Loss 为Alpha-CIOU_Loss,有效提升预测框的回归精度.实验证明,在满足实时性要求的前提下,相比于原始YOLOv5网络,文中方法在海滩小目标游客检测上查准率提升2.00%,查全率提升5.33%,平均精度均值提升4.36%.文中方法在海滩小目标游客密集、遮挡、目标更小的情况下具有更好的鲁棒性.
Keyword:
Reprint 's Address: