计算机工程与科学 ›› 2025, Vol. 47 ›› Issue (9): 1619-1627.
高顺强1,王智文1,白云2
GAO Shunqiang1,WANG Zhiwen1,BAI Yun2
摘要: 针对无监督行人再识别模型计算成本高、模型参数量大的问题,提出一种无监督行人再识别的轻量化特征提取网络。首先,重新设计Ghost Bottleneck,实现模型参数量的压缩,并将ECA注意力模块嵌入到轻量级骨干网络中以提高性能,加强网络的特征提取能力,解决因轻量化而导致的特征丢失问题。其次,引入了集群级动态内存字典和动量更新策略,解决无监督聚类特征的嵌入,有助于缓解特征不一致问题。最后,在数据集LUPerson上进行预训练。在常用的Market-1501,MSMT17和PersonX等公共数据集上开展了大量实验验证。与PPLR,Cluster Contrast和RTMem等方法训练的模型的比较结果表明,LwFEN使模型的参数量下降了24.3%,计算量(以FLOPs衡量)下降了28.12%,并将模型的mAP提升至83.4%。
轻量级网络;无监督行人再识别;动态内存字典;动量更新