基于掩膜引导双流网络的行人重识别方法及装置
摘要:
本发明公开了一种基于掩膜引导双流网络的行人重识别方法及装置,涉及图像识别领域,构建行人重识别模型、第一教师模型和第二教师模型,行人重识别模型包括轮廓单元和暴露单元,将轮廓空间注意力模块作为第一学生模型并与第一教师模型进行蒸馏学习,构造轮廓流中由掩码引导的注意力损失;将暴露空间注意力模块作为第二学生模型并与第二教师模型进行蒸馏学习,构造暴露流中由掩码引导的注意力损失;构造面部信息损失和REID损失;根据轮廓流中由掩码引导的注意力损失、暴露流中由掩码引导的注意力损失、面部信息损失和REID损失构造总损失函数并对行人重识别模型进行训练,得到经训练的行人重识别模型以进行行人重识别,解决背景和衣服因素的干扰问题。
0/0