基于无参注意力和联合损失的行人重识别

摘要

现阶段行人重识别一般只考虑二维特征，将各个特征点统一处理，存在特征提取不足的问题，故提出基于无参注意力的行人重识别（PFNet）来解决上述问题．该模型在Res Net-50网络上进行改进，分别在第一个残差块和第三个残差块后引入无参注意力机制，该注意力机制能根据图片本身特点赋予各特征点合适的权重，可以保留更丰富的信息特征且不会引入额外参数．接着使用自适应平均池化层保留主要特征且捕捉特定域的判别特征，然后使用ID损失、三元组损失和自适应加权排序损失的联合损失函数来训练模型．算法在Market-1501、DukeMTMC-reID和CUHK03三个主流的行人重识别数据集上的首位命中率分别达到95.5%、90.9%和84.3%，平均精度均值分别达到89.6%、81.6%和82.0%．实验结果表明，使用注意力和联合损失函数的策略可以提高模型精度．

关键词

行人重识别 / 残差网络 / 无参注意力 / 联合损失函数