[发明专利]一种基于能量模型的行人重识别方法

权利要求书

1.一种基于能量模型的行人重识别方法，其特征在于，该方法包括如下步骤：

步骤S1：通过摄像头采集城市各路口行人图像N*M个，其中N为行人个数，M为每个行人采集的图像数据，以此作为训练数据集；

步骤S2：利用卷积神经网络训练行人分类模型，设x为某行人图像，f(x)为卷积神经网络所提取的特征，分类模型所训练的损失函数采用基于能量的损失函数；

步骤S3：根据S2中所训练的模型，提取待识别的行人图像特征，并和数据库中的所存储的N个不同行人图像特征比对，以确定待识别的行人身份；

步骤S4：跟S3中的检索结果，通过可视化排序，以确定所识别行人的潜在相似目标和身份；

所述步骤2中基于能量模型的损失函数建立步骤为：

2.1通过卷积神经网络提取行人图像x的特征f(x)；

2.2计算步骤2.1中行人图像特征的目标能量和非目标能量；

E(x,t)＝-f_t(x)

其中，t是目标标签，E(x,t)表示图像的目标能量，E(x,n)表示图像的非目标能量，T为温度常数；

2.3定义基于能量的损失函数

其中max为取最大值函数，α控制损失函数的取值，β控制能量梯度在非目标分量上的分布；最后，根据上述公式作为神经网络训练的损失函数，取代传统softmax损失函数实现网络的分类训练；

在步骤3中带检索图像和数据库中的行人图像相似度的比对方法为：

d(x,x_g)＝|f(x)-f(x_g)|

其中x,x_g分别为待检索的行人图像和数据库中的图像,f(x)和f(x_g)为神经网络所提取的行人特征，d(x,x_g)为图像之间的相似度度量。

2.如权利要求1所述的基于能量模型的行人重识别方法，其特征在于，

在步骤4中，将需要重识别的图片输入重识别网络得到描述特征，将图像库的图片也送入重识别网络得到描述特征，对比描述特征进行排序，将排序结果传递到行人重识别可视化软件显示。