Cross-modality Person Re-identification Framework Based on Improved Hard Triplet Loss

LI Hao,TANG Min,LIN Jian-wu,ZHAO Yun-bo
DOI: https://doi.org/10.11896/jsjkx.191100061
2020-01-01
Computer Science
Abstract:In order to improve the recognition accuracy of cross-modality person re-identification, a feature learning framework based on improved hard triplet loss is proposed. Firstly, traditional hard triplet loss is converted to a global one. Secondly, intra-modality and cross-modality triplet losses are designed to match the global one for model training based on the intra-modality and cross-modality variations. On the basis of improving the hard triplet loss, for the first time the attribute features are designed to increase the ability of the model to extract features in the cross-modality person re-identification model. Finally, for the category imbalance problem, Focal Loss is used to replace the traditional Cross Entropy loss for model training. Compared with existing algorithms, the proposed approach behaves the best on the publicly available RegDB dataset, with an increase of 1.9%-6.4% in all evaluation indicators. In addition, ablation experiments also show that all the three methods can improve the feature extraction of the model.
What problem does this paper attempt to address?