有效

一种基于对比学习的无监督行人重识别方法

李浥东、李志强、曹原周汉、陈乃月、张慧、金一
北京交通大学
李浥东机构 暂无
技术领域 暂无
李志强机构 暂无
技术领域 暂无
曹原周汉机构 暂无
技术领域 暂无
陈乃月机构 暂无
技术领域 暂无
张慧机构 暂无
技术领域 暂无
金一机构 暂无
技术领域 暂无

摘要

本发明提供一种基于对比学习的无监督行人重识别方法,包括:使用ImageNet预训练的ResNet‑50网络来初始化骨干编码器f<Sub>θ</Sub>;利用初始化后的编码器从原始数据集提取特征向量;使用DBSCAN聚类算法进行聚类,并根据每个聚类集群中的向量初始化内存字典;迭代抽取小批量的特征向量结合当前内存字典利用集群对比损失以及离群值损失训练编码器;更新内存字典中相应集群所对应的特征向量;迭代进行从DBSCAN聚类到训练模型编码器之间的步骤,直到模型收敛。本发明提供的方法,通过对比学习迭代提升网络模型的编码能力,使得实际相似的输入图片在高维空间中距离相近,反之差异越大的图片在高维空间中有着越大的距离;具备这样能力的编码器在执行下游任务时,可以快速收敛并且取得更好的性能。

暂无引用专利