结合YOLOv5和质心匹配的轨迹追踪方法
DOI:
CSTR:
作者:
作者单位:

南京师范大学 计算机与电子信息学院/人工智能学院 南京 210023

作者简介:

通讯作者:

中图分类号:

TP183;TP391.4

基金项目:

江苏省自然科学基金(BK20201370)项目资助


Trajectory tracking method combining YOLO v5 and centroid matching
Author:
Affiliation:

School of computer and electronic information / School of artificial intelligence, Nanjing Normal University, Nanjing 210023, China

Fund Project:

  • 摘要
  • |
  • 图/表
  • |
  • 访问统计
  • |
  • 参考文献
  • |
  • 相似文献
  • |
  • 引证文献
  • |
  • 资源附件
  • |
  • 文章评论
    摘要:

    针对施工场所中目标遮挡引起的安全帽佩戴者的轨迹追踪困难的问题,本文提出了一种结合YOLOv5和质心匹配算法的安全帽佩戴检测及轨迹追踪方法。该方法首先采用YOLOv5网络准确检测未佩戴安全帽的人员,计算其质心坐标;进一步的采用扩展卡尔曼滤波器预测目标位置信息;最后采用基于马氏距离及直方图相关性的质心匹配关联算法,结合预测信息实现了目标遮挡环境中的目标轨迹异常修正,可获得准确的目标轨迹。实验结果表明,本方法有效解决了目标跟踪中由目标遮挡引起的目标互换和丢失等问题,在自建数据集中获得了高于传统算法10%以上的目标跟踪准确度,为智慧工地的发展提供了有力的技术支持。

    Abstract:

    Aiming at the difficulty of trajectory tracking of helmet wearers caused by target occlusion in construction sites, a helmet wearing detection and trajectory tracking method combining YOLO v5 and centroid matching algorithm is proposed in this paper. Firstly, YOLO v5 network is used to precisely detect the personnel who do not wear safety helmets and calculate their centroid coordinates. Further, the extended Kalman filter is used to predict the target position information. Finally, the centroid matching association algorithm based on Mahalanobis distance and histogram correlation is adopted. Combined with the prediction information, the target trajectory anomaly correction in the target occlusion environment is realized, and the accurate target trajectory can be obtained. The experimental results show that the proposed method effectively solves the problems of target exchange and loss caused by target occlusion in target tracking, and obtains more than 10% target tracking accuracy higher than the traditional algorithm in the self-built data set, It provides strong technical support for the development of smart construction sites.

    参考文献
    相似文献
    引证文献
引用本文

许玥,宋远伟,赵华.结合YOLOv5和质心匹配的轨迹追踪方法[J].电子测量技术,2022,45(13):123-129

复制
分享
文章指标
  • 点击次数:
  • 下载次数:
  • HTML阅读次数:
  • 引用次数:
历史
  • 收稿日期:
  • 最后修改日期:
  • 录用日期:
  • 在线发布日期: 2024-04-11
  • 出版日期:
文章二维码