基于改进损失函数的多阶段行人属性识别方法

作者：郑少飞; 汤进^*; 罗斌; 王逍; 王文中

来源：模式识别与人工智能, 2018, 31(12): 1085-1095.

DOI：10.16451/j.cnki.issn1003-6059.201812003

• 视频监控
• 行人属性
• 深度学习
• 多阶段
• 损失函数

大量研究工作通过挖掘属性间的正相关性提高视频监控场景下的行人属性识别性能,但对属性间负相关性的探索仍存在不足.为此,文中基于深度学习理论提出多阶段行人属性识别方法,同时探索属性间的正、负相关性.第一阶段计算每个属性在训练过程中的损失值和正确率.第二阶段为平均损失较大且正确率较小的属性单独建立一个网络分支,其它属性仍保留在原分支上,然后两个分支联合预测所有属性.第三阶段新建两个网络分支,结构与第二阶段的分支相同,优化新分支的参数,使其属性识别性能优于第二阶段.最终使用第三阶段的模型进行属性预测.此外,构建增大正负样本差异的改进损失函数,应用于三个阶段的训练,进一步提升模型性能.在两个行人属性识别数据集RAP和PETA上的实验表明,文中方法性能较优.

• 单位
  安徽大学