Few-shot Object Counting and Detection
This work addresses a new task for computer vision researchers and practitioners, enabling counting and detection of objects with minimal supervision, but it is incremental as it builds on existing few-shot methods.
The paper tackles the problem of few-shot object counting and detection by introducing a two-stage training strategy and an uncertainty-aware detector called Counting-DETR, achieving large-margin improvements over strong baselines in both counting and detection metrics on new datasets FSCD-147 and FSCD-LVIS.
We tackle a new task of few-shot object counting and detection. Given a few exemplar bounding boxes of a target object class, we seek to count and detect all objects of the target class. This task shares the same supervision as the few-shot object counting but additionally outputs the object bounding boxes along with the total object count. To address this challenging problem, we introduce a novel two-stage training strategy and a novel uncertainty-aware few-shot object detector: Counting-DETR. The former is aimed at generating pseudo ground-truth bounding boxes to train the latter. The latter leverages the pseudo ground-truth provided by the former but takes the necessary steps to account for the imperfection of pseudo ground-truth. To validate the performance of our method on the new task, we introduce two new datasets named FSCD-147 and FSCD-LVIS. Both datasets contain images with complex scenes, multiple object classes per image, and a huge variation in object shapes, sizes, and appearance. Our proposed approach outperforms very strong baselines adapted from few-shot object counting and few-shot object detection with a large margin in both counting and detection metrics. The code and models are available at https://github.com/VinAIResearch/Counting-DETR.