通过循环挖掘细粒度结构特征,提升任意类别姿态估计精度
Recurrent Feature Mining and Keypoint Mixup Padding for Category-Agnostic Pose Estimation
- 用可变形注意力机制循环提取支持图与查询图的细粒度结构特征
- 在MP-100数据集上相比最先进方法提升3.2%[email protected]
- 用关键点混合法替代零填充,增强不同类别监督信号
类别无关姿态估计旨在根据少量标注的支持图像,在查询图像中定位关键点,适用于任意新类别。现有方法通常通过热图池化提取支持特征,并通过交叉注意力获取支持与查询之间的交互特征,但忽略了从支持和查询图像中挖掘细粒度且结构感知(FGSA)特征,而这些特征对像素级关键点定位至关重要。为此,本文提出一种新颖而简洁的框架,通过循环方式从支持和查询图像中挖掘FGSA特征。具体而言,设计了一种基于可变形注意力机制的FGSA挖掘模块:一方面,在多尺度特征图上应用可变形注意力头以提取细粒度特征;另一方面,通过将关键点的参考点偏移至其连接的关键点来挖掘结构感知特征。借助该模块,反复挖掘支持与查询图像中的FGSA特征,从而获得更优的支持特征和查询估计结果。此外,提出使用混合法生成关键点以统一各类别的关键点数量,相比现有工作采用的零填充能提供更丰富的监督信号。我们在大规模MP-100数据集上进行了大量实验和深入分析,显著超越当前最优方法(+3.2% [email protected])。代码已公开于 https://github.com/chenbys/FMMP。
原文摘要 · Abstract (English)
Category-agnostic pose estimation aims to locate keypoints on query images according to a few annotated support images for arbitrary novel classes. Existing methods generally extract support features via heatmap pooling, and obtain interacted features from support and query via cross-attention. Hence, these works neglect to mine fine-grained and structure-aware (FGSA) features from both support and query images, which are crucial for pixel-level keypoint localization. To this end, we propose a novel yet concise framework, which recurrently mines FGSA features from both support and query images. Specifically, we design a FGSA mining module based on deformable attention mechanism. On the one hand, we mine fine-grained features by applying deformable attention head over multi-scale feature maps. On the other hand, we mine structure-aware features by offsetting the reference points of keypoints to their linked keypoints. By means of above module, we recurrently mine FGSA features from support and query images, and thus obtain better support features and query estimations. In addition, we propose to use mixup keypoints to pad various classes to a unified keypoint number, which could provide richer supervision than the zero padding used in existing works. We conduct extensive experiments and in-depth studies on large-scale MP-100 dataset, and outperform SOTA method dramatically (+3.2\%[email protected]). Code is avaiable at https://github.com/chenbys/FMMP.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。