TriHard Loss Based Multi-Task Person Re-identification

被引:0
|
作者
Chen Q. [1 ]
Chen Y. [1 ]
机构
[1] Key Laboratory of Advanced Process Control for Light Industry, Ministry of Education, Jiangnan University, Wuxi
来源
Jisuanji Fuzhu Sheji Yu Tuxingxue Xuebao/Journal of Computer-Aided Design and Computer Graphics | 2019年 / 31卷 / 07期
关键词
Multi-task network; Pedestrian attributes; Person re-identification; Triplet loss;
D O I
10.3724/SP.J.1089.2019.17463
中图分类号
学科分类号
摘要
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. © 2019, Beijing China Science Journal Publishing Co. Ltd. All right reserved.
引用
收藏
页码:1156 / 1165
页数:9
相关论文
共 29 条
  • [1] Zheng Z.D., Zheng L., Yang Y., A discriminatively learned CNN embedding for person re-identification, ACM Transactions on Multimedia Computing Communications and Applications, 14, 1, (2018)
  • [2] Varior R.R., Haloi M., Wang G., Gated Siamese convolutional neural network architecture for human re-identification, Proceedings of the European Conference on Computer Vision, pp. 791-808, (2016)
  • [3] Cheng D., Gong Y.H., Zhou S.P., Et al., Person re-identification by multi-channel parts-based CNN with improved triplet loss function, Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, pp. 1335-1344, (2016)
  • [4] Chen W.H., Chen X.T., Zhang J.G., Et al., Beyond triplet loss: a deep quadruplet network for person re-identification, Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, pp. 403-412, (2017)
  • [5] Hermans A., Beyer L., Leibe B., In defense of the triplet loss for person re-identification
  • [6] Xiao Q.Q., Luo H., Zhang C., Margin sample mining loss: a deep learning based method for person re-identification
  • [7] An L., Chen X.J., Kafai M., Et al., Improving person re-identification by soft biometrics based re-ranking, Proceedings of the 7th International Conference on Distributed Smart Cameras, pp. 1-6, (2013)
  • [8] Layne R., Hospedales T., Gong S.G., Re-id: hunting attributes in the wild, Proceedings of British Machine Vision Conference, (2014)
  • [9] Layne R., Hospedales T., Gong S.G., Person re-identification by attributes, Proceedings of British Machine Vision Conference, (2012)
  • [10] Su C., Yang F., Zhang S.L., Et al., Multi-task learning with low rank attribute embedding for person re-identification, Proceedings of the IEEE International Conference on Computer Vision, pp. 3739-3747, (2015)