Margin CosReid Network for Pedestrian Re-Identification

• Main Authors: Xiao Yun, Min Ge, Yanjing Sun, Kaiwen Dong, Xiaofeng Hou
• Format: Article
• Language: English
• Published: MDPI AG 2021-02-01
• Series: Applied Sciences
• Subjects: pedestrian re-identification; CNN; softmax loss
• Online Access: https://www.mdpi.com/2076-3417/11/4/1775

This paper proposes a margin CosReid network for effective pedestrian re-identification. Aiming to overcome the overfitting, gradient explosion, and loss function non-convergence problems caused by traditional CNNs, the proposed GBNeck model can realize a faster, stronger generalization, and more discriminative feature extraction task. Furthermore, to enhance the classification ability of the softmax loss function within classes, the margin cosine softmax loss (MCSL) is proposed through a boundary margin introduction to ensure intraclass compactness and interclass separability of the learning depth features and thus to build a stronger metric-based learning model for pedestrian re-identification. The effectiveness of the margin CosReid network was verified on the mainstream datasets Market-1501 and DukeMTMC-reID compared with other state-of-the-art pedestrian re-identification methods.