高级检索
陈巧媛, 陈莹. 基于困难样本三元组损失的多任务行人再识别[J]. 计算机辅助设计与图形学学报, 2019, 31(7): 1156-1165. DOI: 10.3724/SP.J.1089.2019.17463
引用本文: 陈巧媛, 陈莹. 基于困难样本三元组损失的多任务行人再识别[J]. 计算机辅助设计与图形学学报, 2019, 31(7): 1156-1165. DOI: 10.3724/SP.J.1089.2019.17463
Chen Qiaoyuan, Chen Ying. TriHard Loss Based Multi-Task Person Re-identification[J]. Journal of Computer-Aided Design & Computer Graphics, 2019, 31(7): 1156-1165. DOI: 10.3724/SP.J.1089.2019.17463
Citation: Chen Qiaoyuan, Chen Ying. TriHard Loss Based Multi-Task Person Re-identification[J]. Journal of Computer-Aided Design & Computer Graphics, 2019, 31(7): 1156-1165. DOI: 10.3724/SP.J.1089.2019.17463

基于困难样本三元组损失的多任务行人再识别

TriHard Loss Based Multi-Task Person Re-identification

  • 摘要: 针对现有行人再识别算法中采用单一外观特征所存在的特征判别力的不足问题,在深度学习框架下提出一种基于困难样本三元组损失的多任务行人再识别网络,以同时学习身份和属性标签的方式获得更多的行人判别信息.首先将预处理后的图片输入预训练的ResNet-50模型中提取行人特征信息;然后将其特征输入所设计的多任务网络,通过最小化身份和属性联合三元组损失调整网络模型参数,共同完成行人身份和属性判别双分支网络的训练;最后用训练好的模型提取行人特征,用于行人再识别任务,同时实现行人属性的判断.在Market-1501和DukeMTMC-reID这2个数据集上的实验证明,文中网络在行人再识别任务中所提取的特征更具有表征力,识别精度优于现有方法,并且能完成属性识别任务.

     

    Abstract: Aiming to enhance features discrimination for person re-identification, a deep multi-task person re-identification network based on TriHard loss is proposed. By learning identity and attributes labels simultaneously, the network can achieve more discriminative information of pedestrians. Firstly, the pre-trained ResNet-50 is loaded to extract pedestrian features of pre-processed images. Secondly, pedestrian features are fed into the designed multi-task network which consists of two branches. The two branches are trained jointly by minimizing combined TriHard loss of identity and attribute. Finally, the trained model is used to extract pedestrian appearances and attributes features. The features are used for person re-identification and attributes recognition. From the experimental results on the Market-1501 dataset and the DukeMTMC-reID dataset, it shows that features extracted from the proposed network are more discriminative. The multitask network achieves higher identification accuracy over the state-of-the-arts, together with the person attributes recognition.

     

/

返回文章
返回