国内刊号:51-1307/TP
国际刊号:1001-9081
发布日期:
作者:祁子梁, 曲寒冰, 赵传虎, 董良, 李博昭, 王长生
单位:1. 河北工业大学 人工智能与数据科学学院, 天津 300401;2. 北京市科学技术研究院 北京市新技术应用研究所, 北京 100035
关键词:行人再识别,孪生网络,双向最大边界,排序损失,卷积神经网络
基金:国家重点研发计划项目(2018YFC08097000,2018YFC0704800,2018YFF0301000);国家自然科学基金资助项目(91746207);北京市科学技术研究院萌芽计划项目(GS201817)。
针对在实际场景中存在的不同行人图像之间比相同行人图像之间更相似所造成的行人再识别准确率较低的问题,提出一种基于孪生网络并结合识别损失和双向最大边界排序损失的行人再识别方法。首先,对在超大数据集上预训练过的神经网络模型进行结构改造,主要是对最后的全连接层进行改造,使模型可以在行人再识别数据集上进行识别判断;其次,联合识别损失和排序损失监督网络在训练集上的训练,并通过正样本对的相似度值减去负样本对的相似度值大于预定阈值这一判定条件,来使得负例图像对之间的距离大于正例图像对之间的距离;最后,使用训练好的神经网络模型在测试集上测试,提取特征并比对特征之间的余弦相似度。在公开数据集Market-1501、CUHK03和DukeMTMC-reID上进行的实验结果表明,所提方法分别取得了89.4%、86.7%、77.2%的rank-1识别率,高于其他典型的行人再识别方法,并且该方法在基准网络结构下最高达到了10.04%的rank-1识别率提升。
来源:2019年第4期
《计算机应用》期刊编辑部