论文标题
HSGNET:具有分层相似性图形网络的对象重新识别
HSGNet: Object Re-identification with Hierarchical Similarity Graph Network
论文作者
论文摘要
对象重新识别方法由骨干网络,特征聚合和损耗功能组成。但是,大多数骨干网络都缺乏处理丰富规模变化和地雷歧视性特征表示的特殊机制。在本文中,我们首先设计一个分层相似性图模块(HSGM),以减少骨干和重新识别网络的冲突。设计的HSGM构建了丰富的层次图,以挖掘全球本地和本地本地之间的映射关系。其次,我们将特征图与每个分层图中的空间和通道方向分配。 HSGM分别应用了从不同位置提取的节点的空间特征和通道特征,并利用节点之间的相似性得分来构造空间和通道相似性图。在HSGM的学习过程中,我们利用可学习的参数重新优化每个位置的重要性,并评估不同节点之间的相关性。第三,我们通过将HSGM嵌入骨干网络中,开发出一种新颖的分层相似性图网络(HSGNET)。此外,HSGM可以轻松地嵌入任何深度的骨干网络中,以提高对象重新识别能力。最后,在三个大规模对象数据集上进行的广泛实验表明,所提出的HSGNET优于最新的对象重新识别方法。
Object re-identification method is made up of backbone network, feature aggregation, and loss function. However, most backbone networks lack a special mechanism to handle rich scale variations and mine discriminative feature representations. In this paper, we firstly design a hierarchical similarity graph module (HSGM) to reduce the conflict of backbone and re-identification networks. The designed HSGM builds a rich hierarchical graph to mine the mapping relationships between global-local and local-local. Secondly, we divide the feature map along with the spatial and channel directions in each hierarchical graph. The HSGM applies the spatial features and channel features extracted from different locations as nodes, respectively, and utilizes the similarity scores between nodes to construct spatial and channel similarity graphs. During the learning process of HSGM, we utilize a learnable parameter to re-optimize the importance of each position, as well as evaluate the correlation between different nodes. Thirdly, we develop a novel hierarchical similarity graph network (HSGNet) by embedding the HSGM in the backbone network. Furthermore, HSGM can be easily embedded into backbone networks of any depth to improve object re-identification ability. Finally, extensive experiments on three large-scale object datasets demonstrate that the proposed HSGNet is superior to state-of-the-art object re-identification approaches.