Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

Universal-Prototype Enhancing for Few-Shot Object Detection

About

Few-shot object detection (FSOD) aims to strengthen the performance of novel object detection with few labeled samples. To alleviate the constraint of few samples, enhancing the generalization ability of learned features for novel objects plays a key role. Thus, the feature learning process of FSOD should focus more on intrinsical object characteristics, which are invariant under different visual changes and therefore are helpful for feature generalization. Unlike previous attempts of the meta-learning paradigm, in this paper, we explore how to enhance object features with intrinsical characteristics that are universal across different object categories. We propose a new prototype, namely universal prototype, that is learned from all object categories. Besides the advantage of characterizing invariant characteristics, the universal prototypes alleviate the impact of unbalanced object categories. After enhancing object features with the universal prototypes, we impose a consistency loss to maximize the agreement between the enhanced features and the original ones, which is beneficial for learning invariant object characteristics. Thus, we develop a new framework of few-shot object detection with universal prototypes ({FSOD}^{up}) that owns the merit of feature generalization towards novel objects. Experimental results on PASCAL VOC and MS COCO show the effectiveness of {FSOD}^{up}. Particularly, for the 1-shot case of VOC Split2, {FSOD}^{up} outperforms the baseline by 6.8% in terms of mAP.

Aming Wu, Yahong Han, Linchao Zhu, Yi Yang• 2021

Related benchmarks

TaskDatasetResultRank
Object DetectionPASCAL VOC (Novel Set 1)--
223
Object DetectionPASCAL VOC Novel Set 3--
175
Object DetectionMS-COCO (val)--
138
Object DetectionPASCAL VOC Set 2 (novel)--
110
Object DetectionPASCAL VOC Novel Set 2
mAP48.3
100
Object DetectionPASCAL VOC (Novel Set 1)
AP50 (shot=1)43.8
71
Object DetectionPASCAL VOC Set 3 (novel)
AP50 (shot=1)35.5
71
Few-shot Object DetectionMS-COCO 30-shot (novel classes)
nAP (Novel)15.6
34
Few-shot Object DetectionMS-COCO 10-shot (novel classes)
nAP11
34
Showing 9 of 9 rows

Other info

Code

Follow for update