【技术实现步骤摘要】
基于Siamese网络结构融合中层特征的行人重识别方法
本专利技术涉及行人重识别方法,尤其涉及一种基于Siamese网络结构融合中层特征的行人重识别方法。
技术介绍
行人重识别是指在跨摄像头的环境中,判断是否存在给定行人图像。该技术在智能视频监控、维护社会治安等方面发挥着重要的作用。然而行人重识别存在着较大的挑战:当姿势以及光照、背景等环境因素变化时,同一行人的外观在不同的监控视频下存在明显的差异,而不同行人的外观在某种情况下会比较相似。近年来,深度学习的方法被广泛应用,相比于传统手工设计方法,深度学习能够取得更好的性能。在跨摄像头下检索给定的行人图像,深度行人重识别模型通常选用ResNet50作为基础网络在有限的数据集上进行优化,产生巨大的参数量,增加了过拟合的风险,降低泛化能力。同时针对跨摄像头行人出现的模态差异,一些基于GAN网络和利用注意力机制的方法被用来关注模态之间突出的特征,但这些方法模型复杂,训练不稳定。
技术实现思路
专利技术目的:本专利技术的目的是提供一种抗过拟合性能好、鲁棒性高 ...
【技术保护点】
1.一种基于Siamese网络融合中层特征的行人重识别方法,其特征在于,包括以下步骤:/n(1)预训练DenseNet121网络;/n(2)调整DenseNet121网络结构,搭建Siamese网络;/n(3)预处理训练集,利用随机擦除增强数据;/n(4)用预处理的训练集训练Siamese网络;/n(5)使用训练好的Siamese网络模型,得到待查询与图像库图像特征;/n(6)用余弦距离计算查询与图像库图像的相似度,将图像库图像按相似度大小排列。/n
【技术特征摘要】
1.一种基于Siamese网络融合中层特征的行人重识别方法,其特征在于,包括以下步骤:
(1)预训练DenseNet121网络;
(2)调整DenseNet121网络结构,搭建Siamese网络;
(3)预处理训练集,利用随机擦除增强数据;
(4)用预处理的训练集训练Siamese网络;
(5)使用训练好的Siamese网络模型,得到待查询与图像库图像特征;
(6)用余弦距离计算查询与图像库图像的相似度,将图像库图像按相似度大小排列。
2.根据权利要求1所述的基于Siamese网络融合中层特征的行人重识别方法,其特征在于:步骤(1)中,将ImageNet数据集作为训练数据集,训练一个DenseNet121网络,使DenseNet121网络具有初始值。
3.根据权利要求1所述的基于Siamese网络融合中层特征的行人重识别方法,其特征在于:步骤(2)中,在DenseNet121中间层后加入SElayer得到中级特征,最后层加入SElayer得到高级特征,将中级特征与高级特征相融合,在调整后的DenseNet121后添加两个全连接层,构...
【专利技术属性】
技术研发人员:张小瑞,陈旋,孙伟,宋爱国,
申请(专利权)人:南京信息工程大学,
类型:发明
国别省市:江苏;32
还没有人留言评论。发表了对其他浏览者有用的留言会获得科技券。