一种基于分类权重等距分布损失模型的行人重识别方法与流程

文档序号:24161152发布日期:2021-03-05 17:03阅读:108来源:国知局
一种基于分类权重等距分布损失模型的行人重识别方法与流程

[0001]
本发明属于计算机视觉和深度学习领域,涉及一种基于分类权重等距分布损失模型的行人重识别方法。


背景技术:

[0002]
行人重识别,也称行人再识别,是计算机视觉领域近年来十分热门的研究课题,其目标是在给定一张监控摄像头视角下的行人图像后,能够在其他监控视角下检索出该行人。行人重识别技术能够克服固定的视角局限,在多个不相关的监控域进行行人检索。随着公共区域(例如学校、商城、街道路口等)中摄像头网络的普及以及高清摄像头的性能提升,行人重识别在智能安防、智慧城市监控、智能社会治安管理与刑侦追查等方面发挥重要作用。
[0003]
深度学习的兴起和大规模数据集的出现,使得基于深度学习的行人重识别技术近年来被广泛研究且得到快速发展。但由于大规模数据采集的困难性,数据集往往呈现出样本不平衡现象,即小部分类别具有大量的样本数而大部分类别的样本数相对较少,这导致训练出来的模型对样本数量较多的类别表现出较好的识别性能,而对样本数量较少的类别的识别性能较差,降低了模型的泛化能力,给行人重识别任务带来较大的困难。
[0004]
近年来有许多研究提出了解决行人重识别任务中样本不平衡问题的方法,其中一种方法是通过结合分类损失和度量损失来约束行人特征在度量空间中的分布。此类方法首先通过深度网络模型从输入的行人图片中提取高维行人特征。之后,分类损失从行人类别信息的角度利用分类器引导网络模型进行学习,它将行人特征转化为与类别相关的概率并使该特征所属类别的概率最大化,由此得到分类损失;度量损失从行人特征在度量空间的分布的角度,对特征进行约束,使得同类特征在度量空间中相近,类别不同的特征远离,由此得到度量损失。在训练阶段,分类损失和度量损失结合,一起用来训练模型。最后,用训练好的模型分别对测试集中的查询集行人图片和候选集行人图片进行特征提取,并使用距离函数计算查询集图片特征与所有候选集图片特征的距离,将距离结果从小到大进行排序,计算并得到重识别结果。现有技术中,对行人特征在度量空间中的约束效果较差,影响行人重识别的性能。


技术实现要素:

[0005]
本发明的目的在于针对现有技术所存在的问题,提出一种基于分类权重等距分布损失模型的行人重识别方法,该方法从约束行人特征在度量空间中的分布的角度出发,对特征所处度量空间进行归一化约束,将分类层中的分类向量视为类别中心,并在特征空间中对所有分类中心进行等距分布约束,引导特征均匀分布,帮助模型学习更具有辨别力的行人特征,从而提高行人重识别的准确率。
[0006]
本发明所采取的技术方案为:一种基于分类权重等距分布损失模型的行人重识别方法,具体包括以下步骤:
[0007]
步骤1、将训练集中的行人图片进行预处理后,输入到卷积神经网络,获得图片的高维特征表达向量f,用于后续的分类学习;
[0008]
步骤2、使用一个去除偏置项的全连接层对行人特征进行分类学习;在进行分类之前,对全连接层的分类权重向量w和训练用的行人特征向量f进行l2归一化处理,分别得到||w||和f||,同时,令||f||=s,s为一常数。再将归一化后的特征输入全连接层进行分类学习,得到分类结果,计算分类损失其中f
i
表示第i个样本的特征,y
i
代表f
i
的真实标签,w
j
代表分类层中第j个列向量,n为批量训练的样本数量,c为训练类别总数;
[0009]
步骤3、将步骤2中分类层中归一化后的每个分类权重向量||w||视为每类特征的中心,计算每个向量到其它所有权重向量之间的余弦距离之和,得到分类权重等距分布损失其中代表两个分类权重向量和之间的角度,ε为避免等距分类损失在训练初期损失值过大的参数;
[0010]
步骤4、将步骤2和步骤3得到的分类损失和分类权重等距分布损失进行结合,得到总的损失函数,即其中λ为等距分布损失的权重值;
[0011]
步骤5、利用步骤4得到的总损失函数训练并优化卷积神经网络,并最终得到训练好的网络模型;
[0012]
步骤6、将测试集的行人图片输入到训练好的网络模型中,并得到对应的特征向量,并对这些特征向量进行l2归一化处理,得到最终的特征向量;将查询集中行人图片的归一化特征向量逐一与候选集中行人图片的归一化特征向量进行欧式距离对比,根据距离结果进行排序,完成行人重识别任务。
[0013]
本发明相对于现有技术,具有如下优点:
[0014]
第一、本发明利用l2归一化操作,将行人特征与全连接层的分类权重向量归一化到超球体上,因此能够有效地度量特征之间、权重向量之间、特征与权重向量之间的距离,同时解决了度量损失与分类损失约束空间不一致的问题,有利于模型的优化和收敛;
[0015]
第二、本发明利用全连接分类层的分类权重作为训练类别的特征中心,因此可使用分类权重等距分布损失来约束权重向量在特征空间中的分布,从而引导行人特征均匀分布在特征空间中,有效解决行人重识别中因样本不平衡问题带来的性能下降,提高了模型的鲁棒性。
[0016]
第三、本发明使用度量损失和分类损失联合训练的方法,利用等距分布损失使不同类别的特征均匀分布的同时,也使用分类损失缩小同类特征之间的距离,使模型能够挖掘并学习到图片中具有分辨力的行人特征,提高行人重识别的精度。
附图说明
[0017]
图1为本发明的实现流程示意图。
[0018]
图2为本发明中权重等距分布损失的示意图。
[0019]
图3为本发明的网络模型结构示意图。
具体实施方式
[0020]
下面结合实施例及附图对本发明作进一步的详细说明,但本发明的实施方式不限于此。
[0021]
本实施例的一种行人重识别方法,其流程如图1所示。具体实现包括以下步骤:
[0022]
步骤1、从行人数据集(如market-1501、duke-mtmc、cuhk03等)中读取训练数据,并对训练数据进行预处理,预处理操作包括调整图片尺寸、水平随机翻转、随机擦除等;采用批量训练的方法对神经网络进行训练,具体批量采集操作为:从训练集的所有训练数据中,随机挑选p个训练类别,并对每个训练类别随机挑选k个训练样本,利用这些数据组成一组批量训练数据,因此一组批量训练数据的样本数量有pxk个,其中p=16,k=4。
[0023]
步骤2、将挑选的行人样本经过预处理后输入到以resnet-50为基准的卷积神经网络中,得到高维特征矩阵,然后依次通过如图2所示中的全局平均池化层(global average pooling layer)、批归一化层(batch normalization layer)、全连接层(fully connected layer)和批归一化层,得到行人特征向量f;
[0024]
步骤3、将输出的行人特征向量通过l2归一化层(l2 normalization layer)之后,得到归一化后的行人特征向量||f||,并令||f||=s=14;
[0025]
步骤4、将用于分类的全连接层中的偏置项移除,同时对全连接层中的分类权重向量w进行初始化和l2归一化操作,得到||w||;将步骤3得到的归一化行人特征向量作为分类层的输入,得到分类结果,然后使用softmax损失计算分类损失。分类损失可表示为其中f
i
表示第i个样本的特征,y
i
代表f
i
的真实标签,w
j
代表分类层中第j个列向量,n为批量训练的样本数量,c为训练类别总数;
[0026]
步骤5、将分类层中的归一化权重向量||w||视为每类的特征中心,分别计算每个权重向量与其他所有权重向量之间的余弦距离之和,并对每个余弦距离之和的总和求平均,得到分类权重等距分布损失其中代表两个分类权重向量和之间的角度,ε为避免等距分类损失在训练初期损失值过大的参数,取值为10-6

[0027]
步骤6、将分类损失和等距分布损失进行结合,得到总的损失函数l=l
nsl
+λl
edl
,其中λ为等距分布损失的权重值,取值为200;
[0028]
步骤7、根据得到的总损失值,计算每个输入的训练样本梯度,并利用反向传播算法更新神经网络的参数,训练网络直至满足迭代次数要求;
[0029]
步骤8、当网络训练完成后,将原网络中用来分类的全连接层及其之后的网络结构移除,则剩下的网络就变成了特征提取器;将测试图片输入到特征提取器种,得到归一化的特征向量||f||,并分别计算查询集中的每个特征向量与候选集中所有特征向量之间的欧式距离,根据得到的距离结果进行排序,并计算rank1和map结果。
[0030]
以上所述,仅为本发明专利发明优选的实施方式,但发明专利的保护范围不局限于此,其他的任何未背离本发明的精神实质与原理下所作的改变、修饰、替代、组合、简化,均应为等效的置换方式,都包含在本发明的保护范围内。
当前第1页1 2 3 
网友询问留言 已有0条留言
  • 还没有人留言评论。精彩留言会获得点赞!
1