arXiv:2409.19899cs.CV2024-09ECCV被引 12

让模型看懂各种新描述,实现零样本关键点检测

OpenKD: Opening Prompt Diversity for Zero- and Few-shot Keypoint Detection

论文配图:OpenKD: Opening Prompt Diversity for Zero- and Few-shot Keypoint Detection
图 1 · 摘自论文原文
  • 用多模态原型集支持图文混合提示
  • 引入插值文本和关键点提升零样本检测能力
  • 结合大语言模型解析复杂指令,适合多场景应用

利用基础模型(如CLIP)构建通用关键点检测器正受关注。现有方法通常仅支持文本提示(如“猫的鼻子”)或视觉提示(带标注的支持图),从而实现零样本或少样本检测。但多模态提示的研究仍不充分,提示在语义和语言上的多样性尚未打开。例如如何处理未见文本提示,以及“你能检测猫的鼻子和耳朵吗?”这类多样化提问。本文从模态、语义(已见/未见)和语言三个维度拓展提示多样性,提出OpenKD模型,通过多模态原型集支持视觉与文本提示。为提升未见文本的推理能力,训练中加入视觉与文本域插值的辅助关键点和文本,增强空间推理,显著提升零样本新关键点检测性能。实验表明,该方法在零-少样本关键点检测上达到领先水平,并开启处理未见文本与多样化提示的新路径。代码与数据见https://github.com/AlanLuSun/OpenKD。

原文摘要 · Abstract (English)

Exploiting the foundation models (e.g., CLIP) to build a versatile keypoint detector has gained increasing attention. Most existing models accept either the text prompt (e.g., ``the nose of a cat''), or the visual prompt (e.g., support image with keypoint annotations), to detect the corresponding keypoints in query image, thereby, exhibiting either zero-shot or few-shot detection ability. However, the research on taking multimodal prompt is still underexplored, and the prompt diversity in semantics and language is far from opened. For example, how to handle unseen text prompts for novel keypoint detection and the diverse text prompts like ``Can you detect the nose and ears of a cat?'' In this work, we open the prompt diversity from three aspects: modality, semantics (seen v.s. unseen), and language, to enable a more generalized zero- and few-shot keypoint detection (Z-FSKD). We propose a novel OpenKD model which leverages multimodal prototype set to support both visual and textual prompting. Further, to infer the keypoint location of unseen texts, we add the auxiliary keypoints and texts interpolated from visual and textual domains into training, which improves the spatial reasoning of our model and significantly enhances zero-shot novel keypoint detection. We also found large language model (LLM) is a good parser, which achieves over 96% accuracy to parse keypoints from texts. With LLM, OpenKD can handle diverse text prompts. Experimental results show that our method achieves state-of-the-art performance on Z-FSKD and initiates new ways to deal with unseen text and diverse texts. The source code and data are available at https://github.com/AlanLuSun/OpenKD.

关键点检测零样本学习多模态提示大语言模型

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。