[发明专利]行人再识别模型训练方法及装置、电子设备和存储介质

说明书

本发明实施例公开了一种行人再识别模型训练方法、装置、电子设备及存储介质，其中的方法包括：获取视频中各个追踪序列；根据所述追踪序列选取未标注数据作为数据样本；对所述数据样本进行特征提取，得到数据样本特征；根据所述数据样本特征，对包括传导质心投影层的深度学习网络进行训练，得到行人再识别模型；其中，所述传导质心投影层是根据所述追踪序列确定的特征向量投影层。本发明实施例可提升模型性能和网络效果。

技术领域

本发明涉及深度学习技术，尤其是一种行人再识别模型训练方法及装置、电子设备和存储介质。

背景技术

行人再识别，用于判断某个摄像头中的某个行人是否曾经出现在其他的摄像头中，即需要将某个行人特征与其他行人特征进行对比，判断是否属于同一个行人。

发明内容

本发明实施例提供一种行人再识别模型训练方法及装置、电子设备和存储介质的技术方案。

本发明实施例提供的一种行人再识别模型训练方法，包括：

获取视频中各个追踪序列；

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

对所述数据样本进行特征提取，得到数据样本特征；

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

在一种可选方式中，所述根据所述追踪序列选取未标注数据作为数据样本包括：

计算视频追踪序列中非同一身份用户的追踪序列的概率；

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

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

在一种可选方式中，所述追踪序列是指视频中同一个用户的所有视频帧中的外接框序列。

在一种可选方式中，还包括：确定所述传导质心投影层；

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

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

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

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

在一种可选方式中，基于半监督学习模式训练得到所述行人再识别模型，或者，基于无监督学习模式训练得到所述行人再识别模型。

在一种可选方式中，当基于半监督学习模式训练所述行人再识别模型时，所述数据样本还包括标注数据；

所述对所述数据样本进行特征提取得到数据样本特征包括：

将所述标注数据以及所述未标注数据进行卷积处理，分别得到标注数据特征和未标注数据特征；其中：所述数据样本特征包括所述标注数据特征和所述未标注数据特征。

在一种可选方式中，当基于无监督学习模式训练所述行人再识别模型时，所述对所述数据样本进行特征提取得到数据样本特征包括：

将所述未标注数据进行卷积处理，得到未标注数据特征；其中：所述数据样本特征为所述未标注数据特征。

本发明实施例还提供一种行人再识别模型训练装置，包括：

追踪序列获取单元，用于获取视频中各个追踪序列；