基于数据增强的赛马场景多目标追踪方法

作者:吴永贤; 刘旭宇; 颜旭立; 田星; 钟灿琨
来源:2022-05-25, 中国, ZL202210573112.0.

摘要

本发明公开了一种基于数据增强的赛马场景多目标追踪方法,包括:1)对原始赛马训练视频、测试视频的每一帧图像进行人工标注,得到原始赛马训练标注视频、测试标注视频;2)采用ACGAN网络对原始赛马训练标注视频进行数据增强,获得GAN赛马标注伪视频;3)采用FairMOT多目标追踪模型对原始赛马训练标注视频和GAN赛马标注伪视频进行训练,得到最终的赛马场景多目标追踪模型;4)输入原始赛马测试视频至赛马场景多目标追踪模型中,得到处理后的赛马追踪测试视频,与原始赛马测试标注视频进行结果比较,判断追踪效果。本发明提高了目标追踪的准确度,缓解赛马场景下在出现遮挡、镜头切换和冲刺模糊现象时的身份ID频繁切换问题,进一步可实现更灵活和精确的下游应用。