一种基于Transformer与轨迹查询组的多目标跟踪方法、装置、设备及介质
申请号:CN202510199521
申请日期:2025-02-24
公开号:CN120279119B
公开日期:2025-11-14
类型:发明专利
摘要
本发明提供了一种基于Transformer与轨迹查询组的多目标跟踪方法、装置、设备及介质,涉及计算机视觉技术领域。包括:将当前图像帧输入待训练的多目标跟踪模型;通过主干网络和Transformer编码器对当前图像帧进行处理得到当前图像编码特征;在上一图像帧中检测到目标的情况下,将当前图像编码特征、检测查询特征和轨迹查询组输入Transformer解码器,得到目标对应的输出嵌入结果;基于目标对应的输出嵌入结果确定样本多目标跟踪结果;基于样本多目标跟踪结果和多目标跟踪结果标签对多目标跟踪模型进行训练,得到训练好的多目标跟踪模型,以用于进行多目标跟踪,从而提升复杂场景中多目标跟踪的跟踪精度。
技术关键词
查询特征
轨迹
跟踪方法
图像编码
视频流
解码器
样本
编码器
记忆
计算机视觉技术
模型训练模块
可读存储介质
内容更新
标签
跟踪装置
处理器
网络
输入模块