Open-Vocabulary Animal Keypoint Detection with Semantic-feature Matching

TL;DR

This paper introduces a zero-shot animal keypoint detection framework that uses text prompts and semantic features, enabling detection across species without prior annotations, outperforming existing methods.

Contribution

The paper proposes the KDSM framework for open-vocabulary keypoint detection, combining vision and language models with novel modules for improved generalization and zero-shot performance.

Findings

KDSM outperforms baseline methods in keypoint detection tasks.

Zero-shot approach achieves results comparable to few-shot methods.

Framework demonstrates strong generalization across animal species.

Abstract

Current image-based keypoint detection methods for animal (including human) bodies and faces are generally divided into full-supervised and few-shot class-agnostic approaches. The former typically relies on laborious and time-consuming manual annotations, posing considerable challenges in expanding keypoint detection to a broader range of keypoint categories and animal species. The latter, though less dependent on extensive manual input, still requires necessary support images with annotation for reference during testing. To realize zero-shot keypoint detection without any prior annotation, we introduce the Open-Vocabulary Keypoint Detection (OVKD) task, which is innovatively designed to use text prompts for identifying arbitrary keypoints across any species. In pursuit of this goal, we have developed a novel framework named Open-Vocabulary Keypoint Detection with Semantic-feature Matching (KDSM). This framework synergistically combines vision and language models, creating an interplay between language features and local keypoint visual features. KDSM enhances its capabilities by integrating Domain Distribution Matrix Matching (DDMM) and other special modules, such as the Vision-Keypoint Relational Awareness (VKRA) module, improving the framework's generalizability and overall performance.Our comprehensive experiments demonstrate that KDSM significantly outperforms the baseline in terms of performance and achieves remarkable success in the OVKD task.Impressively, our method, operating in a zero-shot fashion, still yields results comparable to state-of-the-art few-shot species class-agnostic keypoint detection methods.We will make the source code publicly accessible.