一种基于YOLO V5的球员追踪检测方法

摘要

本发明涉及一种基于YOLO V5的球员追踪检测方法。利用K‑means聚类方法对目标框的参数进行确定及对数据增强的数据集进行训练，利用YOLO V5和DeepSort算法，并且将prediction heads集成到YOLO V5中，在高密度场景中准确定位和追踪目标，并用无人机在球场上空进行场景实时捕获。本发明能做到较高的模型精度，能对目标物和重叠遮挡目标进行准确度高的识别，能够记录一整场比赛中运动员的运动轨迹和细节，帮助我们更好的对运动员和其所在球队进行数据分析，比如运动员在什么区域得分较多，球员喜欢在什么位置持球等等，帮助教练和运动员赛后更好复盘，对运动员进行有针对性的指导和训练和对对手球队球员进行针对性的防守部署；对于用户，有助于用户快速找到感兴趣的体育视频节目或片段。