CountGD++: Generalized Prompting for Open-World Counting
About
The flexibility and accuracy of methods for automatically counting objects in images and videos are limited by the way the object can be specified. While existing methods allow users to describe the target object with text and visual examples, the visual examples must be manually annotated inside the image, and there is no way to specify what not to count. To address these gaps, we introduce novel capabilities that expand how the target object can be specified. Specifically, we extend the prompt to enable what not to count to be described with text and/or visual examples, introduce the concept of `pseudo-exemplars' that automate the annotation of visual examples at inference, and extend counting models to accept visual examples from both natural and synthetic external images. We also use our new counting model, CountGD++, as a vision expert agent for an LLM. Together, these contributions expand the prompt flexibility of multi-modal open-world counting and lead to significant improvements in accuracy, efficiency, and generalization across multiple datasets. Code is available at https://github.com/niki-amini-naieni/CountGDPlusPlus.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Crowd Counting | ShanghaiTech Part B | MAE28 | 160 | |
| Crowd Counting | ShanghaiTech Part A | MAE116 | 138 | |
| Object Detection | FSCD-147 (test) | AP38.93 | 29 | |
| Counting | FSCD-147 (test) | MAE8.39 | 21 | |
| Object Counting | PairTally (test) | MAE35.27 | 7 | |
| Object Counting | Blood Cell Detection (test) | MAE1.52 | 6 | |
| Object Counting | OmniCount Fruits (test) | MAE0.56 | 6 | |
| Object Detection | Blood Cell Detection (test) | AP54 | 5 | |
| Object Detection | OmniCount Fruits (test) | AP69 | 5 | |
| Object Counting | PrACo (test) | NMN0.88 | 5 |