用图文描述替代标注图像,实现无需支持样本的通用姿态估计
CapeLLM: Support-Free Category-Agnostic Pose Estimation with Multimodal Large Language Models
- 仅需查询图像和文本描述,不依赖标注支持图
- 在MP-100数据集上1次、5次样本下均达新最优
- 利用大模型先验知识,自适应推理未知关键点位置
传统无类别姿态估计依赖带标注关键点的支持图像,过程繁琐且难以覆盖多样物体。近期方法尝试使用文本查询以提升稳定性和泛化能力,但仍受限于对支持查询的依赖、未充分利用预训练大语言模型中的丰富先验,以及参数化分布假设的局限。为此,我们提出CapeLLM,首个专为无类别姿态估计设计的多模态大语言模型。该方法仅以查询图像和详细文本描述作为输入,即可估计无类别关键点。通过有效的训练策略与精心设计的指令,将多模态大语言模型应用于姿态估计。此外,我们提出一种推理机制,可灵活建模关键点的空间分布与不确定性,基于上下文线索进行自适应优化。我们在多个场景下进行了广泛实验,不仅关注模型架构与提示设计,还确保对输入变化的鲁棒性。所提方法在MP-100基准上实现了1次及5次样本设置下的新最优性能,显著推进了无类别姿态估计的发展。代码已开源。
原文摘要 · Abstract (English)
Category-agnostic pose estimation (CAPE) has traditionally relied on support images with annotated keypoints, a process that is often cumbersome and may fail to fully capture the necessary correspondences across diverse object categories. Recent efforts have explored the use of text queries, leveraging their enhanced stability and generalization capabilities. However, existing approaches often remain constrained by their reliance on support queries, their failure to fully utilize the rich priors embedded in pre-trained large language models, and the limitations imposed by their parametric distribution assumptions. To address these challenges, we introduce CapeLLM, the first multimodal large language model (MLLM) designed for CAPE. Our method only employs query image and detailed text descriptions as an input to estimate category-agnostic keypoints. Our method encompasses effective training strategies and carefully designed instructions for applying the MLLM to CAPE. Moreover, we propose an inference mechanism that further enhances the reasoning process for unseen keypoints. while flexibly modeling their underlying spatial distribution and uncertainty, allowing for adaptive refinement based on contextual cues. We conducted extensive experiments to apply the MLLM to CAPE effectively, focusing not only on the model architecture and prompt design but also on ensuring robustness across input variations. Our approach sets a new state-of-the-art on the MP-100 benchmark in the 1-shot and even 5-shot setting, marking a significant advancement in the field of category-agnostic pose estimation. Code is available at https://github.com/Junhojuno/CapeLLM.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。