Detect Any Keypoints: An Efficient Light-Weight Few-Shot Keypoint Detector
>The 38th Annual AAAI Conference on Artificial Intelligence (AAAI 2024)
>Changsheng Lu, Piotr Koniusz
>In order to maintain flexibility of detecting varying number of keypoints, existing few-shot keypoint detection (FSKD) approaches modulate query feature map per support keypoint, then detect the corresponding keypoint from each modulated feature via a detection head. Such modulation-detection separate design would scale up model into heavy yet slow one when the number of keypoints increases. To overcome this issue, we design a novel light-weight detector which combines modulation and detection into one step, greatly reducing the computational cost without sacrifice of performance. Moreover, to bridge the large domain shift of keypoints between seen and unseen species, we further improve our model with mean feature based contrastive learning to align keypoint distributions that encourage better keypoint representations for FSKD.