多目标行人轨迹预测模型训练方法、预测方法及装置

基本信息

申请号 CN202210244026.5 申请日 -
公开(公告)号 CN114419105B 公开(公告)日 2022-07-15
申请公布号 CN114419105B 申请公布日 2022-07-15
分类号 G06T7/246(2017.01)I;G06V40/10(2022.01)I;G06V40/20(2022.01)I 分类 计算;推算;计数;
发明人 周波;蔡芳发;苗瑞;邹小刚 申请(专利权)人 深圳海清智元科技股份有限公司
代理机构 北京同立钧成知识产权代理有限公司 代理人 -
地址 518100广东省深圳市宝安区西乡街道固戍开发区泰华梧桐工业园8号建筑3层
法律状态 -

摘要

摘要 本申请提供一种多目标行人轨迹预测模型训练方法、预测方法及装置,方法包括:将预先建立的多目标行人轨迹预测模型置于多任务学习框架下,该多目标行人轨迹预测模型包括多目标行人检测模型以及多目标行人跟踪模型,采用预先存储的训练数据对多目标行人检测模型以及多目标行人跟踪模型进行联合训练,获得训练后的多目标行人轨迹预测模型。在本申请中,在多任务学习框架下对多目标行人跟踪模型以及多目标行人检测模型进行联合训练,可以在训练过程中实现参数共享,从而可以提高模型的训练效果,进而提高多目标行人轨迹预测模型的泛化性和精确度。