Journal of Beijing University of Posts and Telecommunications

  • EI核心期刊

Journal of Beijing University of Posts and Telecommunications ›› 2021, Vol. 44 ›› Issue (3): 67-72.doi: 10.13190/j.jbupt.2020-147

Previous Articles     Next Articles

Person Re-Identification Method Based on Image Style Transfer

WANG Chen-kui1, CHEN Yue-lin1, CAI Xiao-dong2   

  1. 1. School of Mechanical and Electrical Engineering, Guilin University of Electronic Technology, Guilin 541000, China;
    2. School of Information and Communication, Guilin University of Electronic Technology, Guilin 541000, China
  • Received:2020-09-08 Online:2021-06-28 Published:2021-06-23

Abstract: The training set of the existing person re-identification model comes from limited fixed collection equipment, and the sample style lacks diversity. Through the cyclic generative adversarial network, the image data captured by different cameras can be styled and style transferred, which can improve the diversity of sample styles at a lower cost. In order to improve the generalization ability of the model, a new training mechanism of positive and negative samples fusion is designed. Firstly, the samples after the style transfer are regarded as negative samples, and the samples before the style transfer are regarded as positive samples. The positive and negative samples are sent to the model training at the same time. Furthermore, in order to prevent over fitting and consider the loss of false labels positions, label smoothing regularization is adopted. At the same time, in order to pay more attention to difficult and error-prone samples, and to optimize the loss of negative samples, a focal loss function is adopted. Experiments show that there is a significant increase of 1.51% and 2.07% on the Market-1501 and DukeMTMC-reID datasets, respectively.

Key words: cyclic generative adversarial network, person re-identification, label smoothing loss regularization, focus loss

CLC Number: