一种基于俯视投射的多目标跟踪网络构建方法及装置
摘要:
本发明公开了一种基于俯视投射的多目标跟踪网络构建方法及装置,方法包括如下步骤:采集连续多个时间帧的训练图像组,每帧训练图像组均包括相同的多个目标,并标注各个目标的类别、中心点坐标、标注框宽高及空间信息,得到多目标跟踪数据集;构建多目标跟踪的初始网络,包括主干网、俯视编码器及空间解码器;根据多目标跟踪数据集对初始网络进行迭代训练至收敛,得到基于俯视投射的多目标跟踪网络。本发明构建基于俯视投射的多目标跟踪网络,缓解克服当前以2D检测实现多目标跟踪时存在的重叠遮挡、精准度不高等问题,提升了多目标跟踪的能力。
0/0