Learning to Count without Annotations

Read original: arXiv:2307.08727 - Published 4/1/2024 by Lukas Knobel, Tengda Han, Yuki M. Asano
Total Score

0

🔄

Sign in to get full access

or

If you already have an account, we'll log you in

The paper introduces UnCounTR, an innovative model that can perform reference-based object counting without relying on manual annotations. Traditional supervised methods require extensive manual labeling, which is costly and time-consuming, especially when dealing with numerous objects in images.

The key idea is to construct Self-Collages, synthetic training images containing various pasted objects. These Self-Collages provide a rich learning signal, covering diverse object types and counts, enabling the model to learn the counting task without manual supervision.

The method leverages existing unsupervised representations and segmentation techniques to achieve reference-based counting capabilities. Experiments demonstrate that UnCounTR outperforms simple baselines and generalizes well to diverse datasets, showcasing the potential of unsupervised learning for this task.

Highlights:

  • UnCounTR: A model for reference-based object counting without manual annotations.
  • Self-Collages: Synthetic training images with pasted objects, providing a rich learning signal.
  • Utilizes unsupervised representations and segmentation techniques.
  • Outperforms baselines and generalizes well across datasets.
  • Demonstrates the feasibility of unsupervised learning for object counting tasks.


This summary was produced with help from an AI and may contain inaccuracies - check out the links to read the original source documents!

Follow @aimodelsfyi on 𝕏 →

Related Papers

🔄

Total Score

0

Learning to Count without Annotations

Lukas Knobel, Tengda Han, Yuki M. Asano

While recent supervised methods for reference-based object counting continue to improve the performance on benchmark datasets, they have to rely on small datasets due to the cost associated with manually annotating dozens of objects in images. We propose UnCounTR, a model that can learn this task without requiring any manual annotations. To this end, we construct Self-Collages, images with various pasted objects as training samples, that provide a rich learning signal covering arbitrary object types and counts. Our method builds on existing unsupervised representations and segmentation techniques to successfully demonstrate for the first time the ability of reference-based counting without manual supervision. Our experiments show that our method not only outperforms simple baselines and generic models such as FasterRCNN and DETR, but also matches the performance of supervised counting models in some domains.

Read more

4/1/2024

AFreeCA: Annotation-Free Counting for All
Total Score

0

AFreeCA: Annotation-Free Counting for All

Adriano D'Alessandro, Ali Mahdavi-Amiri, Ghassan Hamarneh

Object counting methods typically rely on manually annotated datasets. The cost of creating such datasets has restricted the versatility of these networks to count objects from specific classes (such as humans or penguins), and counting objects from diverse categories remains a challenge. The availability of robust text-to-image latent diffusion models (LDMs) raises the question of whether these models can be utilized to generate counting datasets. However, LDMs struggle to create images with an exact number of objects based solely on text prompts but they can be used to offer a dependable textit{sorting} signal by adding and removing objects within an image. Leveraging this data, we initially introduce an unsupervised sorting methodology to learn object-related features that are subsequently refined and anchored for counting purposes using counting data generated by LDMs. Further, we present a density classifier-guided method for dividing an image into patches containing objects that can be reliably counted. Consequently, we can generate counting data for any type of object and count them in an unsupervised manner. Our approach outperforms other unsupervised and few-shot alternatives and is not restricted to specific object classes for which counting data is available. Code to be released upon acceptance.

Read more

8/6/2024

💬

Total Score

0

TFCounter:Polishing Gems for Training-Free Object Counting

Pan Ting, Jianfeng Lin, Wenhao Yu, Wenlong Zhang, Xiaoying Chen, Jinlu Zhang, Binqiang Huang

Object counting is a challenging task with broad application prospects in security surveillance, traffic management, and disease diagnosis. Existing object counting methods face a tri-fold challenge: achieving superior performance, maintaining high generalizability, and minimizing annotation costs. We develop a novel training-free class-agnostic object counter, TFCounter, which is prompt-context-aware via the cascade of the essential elements in large-scale foundation models. This approach employs an iterative counting framework with a dual prompt system to recognize a broader spectrum of objects varying in shape, appearance, and size. Besides, it introduces an innovative context-aware similarity module incorporating background context to enhance accuracy within messy scenes. To demonstrate cross-domain generalizability, we collect a novel counting dataset named BIKE-1000, including exclusive 1000 images of shared bicycles from Meituan. Extensive experiments on FSC-147, CARPK, and BIKE-1000 datasets demonstrate that TFCounter outperforms existing leading training-free methods and exhibits competitive results compared to trained counterparts.

Read more

5/7/2024

Iterative Object Count Optimization for Text-to-image Diffusion Models
Total Score

0

Iterative Object Count Optimization for Text-to-image Diffusion Models

Oz Zafar, Lior Wolf, Idan Schwartz

We address a persistent challenge in text-to-image models: accurately generating a specified number of objects. Current models, which learn from image-text pairs, inherently struggle with counting, as training data cannot depict every possible number of objects for any given object. To solve this, we propose optimizing the generated image based on a counting loss derived from a counting model that aggregates an object's potential. Employing an out-of-the-box counting model is challenging for two reasons: first, the model requires a scaling hyperparameter for the potential aggregation that varies depending on the viewpoint of the objects, and second, classifier guidance techniques require modified models that operate on noisy intermediate diffusion steps. To address these challenges, we propose an iterated online training mode that improves the accuracy of inferred images while altering the text conditioning embedding and dynamically adjusting hyperparameters. Our method offers three key advantages: (i) it can consider non-derivable counting techniques based on detection models, (ii) it is a zero-shot plug-and-play solution facilitating rapid changes to the counting techniques and image generation methods, and (iii) the optimized counting token can be reused to generate accurate images without additional optimization. We evaluate the generation of various objects and show significant improvements in accuracy. The project page is available at https://ozzafar.github.io/count_token.

Read more

8/22/2024