利索能及
我要发布
收藏
专利号: 201810152276X
申请人: 北京市商汤科技开发有限公司
专利类型:发明专利
专利状态:已下证
更新日期:2026-04-09
缴费截止日期: 暂无
联系人

摘要:

权利要求书:

1.一种行人再识别模型训练方法,其特征在于,包括:获取视频中各个追踪序列;

根据所述追踪序列选取未标注数据作为数据样本;

对所述数据样本进行特征提取,得到数据样本特征;

根据所述数据样本特征,对包括传导质心投影层的深度学习网络进行训练,得到行人再识别模型;其中,所述传导质心投影层是根据所述追踪序列确定的特征向量投影层。

2.根据权利要求1所述的方法,其特征在于,所述根据所述追踪序列选取未标注数据作为数据样本包括:计算视频追踪序列中非同一身份用户的追踪序列的概率;

在所述非同一身份用户的追踪序列的概率大于预置的追踪序列概率阈值的条件下,确定出追踪序列数量以及追踪序列选取时间间隔,并确定图像帧选取时间间隔;

根据所述追踪序列数量、所述追踪序列选取时间间隔以及所述图像帧选取时间间隔,从视频帧中选择出所述未标注数据。

3.根据权利要求1所述的方法,其特征在于,所述追踪序列是指视频中同一个用户的所有视频帧中的外接框序列。

4.根据权利要求1所述的方法,其特征在于,还包括:确定所述传导质心投影层;

所述确定所述传导质心投影层包括:

计算所述追踪序列中每一帧在所述深度学习网络的特征空间中的特征向量平均值;

确定所述特征向量平均值为质心;

将每一帧的特征向量与所述质心进行内积计算,将得到的质心投影确定为所述传导质心投影层。

5.根据权利要求1-4任一项所述的方法,其特征在于,基于半监督学习模式训练得到所述行人再识别模型,或者,基于无监督学习模式训练得到所述行人再识别模型。

6.根据权利要求5所述的方法,其特征在于,当基于半监督学习模式训练所述行人再识别模型时,所述数据样本还包括标注数据;

所述对所述数据样本进行特征提取得到数据样本特征包括:将所述标注数据以及所述未标注数据进行卷积处理,分别得到标注数据特征和未标注数据特征;其中:所述数据样本特征包括所述标注数据特征和所述未标注数据特征。

7.根据权利要求5所述的方法,其特征在于,当基于无监督学习模式训练所述行人再识别模型时,所述对所述数据样本进行特征提取得到数据样本特征包括:将所述未标注数据进行卷积处理,得到未标注数据特征;其中:所述数据样本特征为所述未标注数据特征。

8.一种行人再识别模型训练装置,其特征在于,包括:追踪序列获取单元,用于获取视频中各个追踪序列;

样本选取单元,用于根据所述追踪序列选取未标注数据作为数据样本;

特征提取单元,用于对所述数据样本进行特征提取,得到数据样本特征;

网络训练单元,用于根据所述数据样本特征,对包括传导质心投影层的深度学习网络进行训练,得到行人再识别模型;其中,所述传导质心投影层是根据所述追踪序列确定的特征向量投影层。

9.一种计算机可读存储介质,其上存储有计算机程序,其特征在于,该程序被处理器执行时实现权利要求1-7任一项所述方法的步骤。

10.一种电子设备,包括存储器、处理器及存储在存储器上并可在处理器上运行的计算机程序,其特征在于,所述处理器执行所述程序时实现权利要求1-7任一项所述方法的步骤。