[发明专利]一种基于生成网络的行人重识别数据生成和扩充方法

摘要

本发明提供了一种基于生成网络的行人重识别数据生成和扩充方法，步骤为：利用视频预测网络生成新的行人视频帧样本。利用深度生成对抗网络进行端到端的行人背景变换数据生成。利用不同的数据生成方法进行行人数据集的广度和丰富性的扩充。将扩充的数据集送入特征提取网络中提取特征并用欧氏距离评估性能。本方法同时考虑了行人的类内和类间数据扩充，联合利用了不同的生成网络生成更多更丰富的样本，扩充的数据集具有很好的多样性和鲁棒性，能更好的解决与适应由于样本数量不足和背景干扰带来的性能损失，具有普遍的适用性，扩充的数据集在下一步的行人识别中能发挥更好的性能及效率。

主权项

1.一种基于生成网络的行人重识别数据生成和扩充方法，其特征在于：包括以下步骤：/n第一步：基于无监督的视频预测生成网络，形成行人视频预测生成网络P；/n第二步：对行人视频中的每一个图像序列s＝(s⁰,...,s^T-1)，均经过第一步中的行人视频预测生成网络P生成新的视频帧序列p＝(p⁰,...,p^T-1)，并同原始视频帧计算生成目标损失函数L并反传进行优化迭代，得到最优模型；其中，s^T-1表示第T帧原始图像，p^T-1表示第T帧预测图像，上标T表示序列长度；/n第三步：在整个待扩充行人数据集上对第二步得到的最优模型进行应用，对待扩充行人数据集中的每一个行人视频帧，优化得到新的预测帧，并加入到待扩充行人数据集中进行类内视频长度的扩充，得到类内生成数据，用于减少由于数据量不足带来的过拟合现象；/n第四步：基于传统生成对抗网络，形成行人图片生成对抗网络N，其中行人图片生成对抗网络N包含主要由卷积网络构成的生成单元G和描述单元D；/n第五步：对待扩充行人数据集中的每一张输入图片s(t)，均经过第四步中的行人图片生成对抗网络N的生成单元G，通过生成单元G生成新的图片o(t)；/n第六步：对待扩充行人数据集中的每一张输入图片s(t)所对应的原始背景替换图y(t)和对应的生成图片o(t)，均经过第四步中的行人图片生成对抗网络N的描述单元D，通过描述单元D中的损失函数L_C(G,D)判别二者的差异损失，并将损失函数L_C(G,D)分别反向传播于生成单元G和描述单元D进行对抗迭代优化；/n第七步：加入一范数损失函数，用于减少生成图片的模糊程度；/n第八步：最大化描述单元D的损失，用于使网络能够区分生成图片和原始图片；最小化生成单元G的损失，用于使生成图片能够接近原始图片；进行第六步的对抗迭代优化，得到最优的端到端的背景替换数据生成模型G^*；/n第九步：应用第八步的背景替换数据生成模型G^*得到新的端到端的背景替换生成图片；将背景替换生成图片加入待扩充行人数据集中进行类间数据扩充，得到类间生成数据，用于增加数据集的多样性，减少背景干扰；/n第十步：利用第三步得到的类内生成数据和第九步得到的类间生成数据对待扩充行人数据集进行扩充，得到最终的扩充数据集，并将最终的扩充数据集送入特征提取网络中提取特征并用欧氏距离评估性能。/n