A Unified Object Counting Network with Object Occupation Prior

TL;DR

This paper introduces a novel unified object counting network capable of handling evolving object classes by leveraging a class-agnostic mask and class-incremental learning, enabling efficient adaptation to new classes without retraining from scratch.

Contribution

It proposes the first evolving object counting dataset and a unified network that incorporates class-agnostic masks and incremental learning to adapt to new object classes.

Findings

The model effectively adapts to new classes with minimal retraining.

Knowledge distillation helps retain previous class information.

The approach outperforms existing methods on the new dataset.

Abstract

The counting task, which plays a fundamental role in numerous applications (e.g., crowd counting, traffic statistics), aims to predict the number of objects with various densities. Existing object counting tasks are designed for a single object class. However, it is inevitable to encounter newly coming data with new classes in our real world. We name this scenario as \textit{evolving object counting}. In this paper, we build the first evolving object counting dataset and propose a unified object counting network as the first attempt to address this task. The proposed model consists of two key components: a class-agnostic mask module and a class-incremental module. The class-agnostic mask module learns generic object occupation prior via predicting a class-agnostic binary mask (e.g., 1 denotes there exists an object at the considering position in an image and 0 otherwise). The class-incremental module is used to handle new coming classes and provides discriminative class guidance for density map prediction. The combined outputs of class-agnostic mask module and image feature extractor are used to predict the final density map. When new classes come, we first add new neural nodes into the last regression and classification layers of class-incremental module. Then, instead of retraining the model from scratch, we utilize knowledge distillation to help the model remember what have already learned about previous object classes. We also employ a support sample bank to store a small number of typical training samples of each class, which are used to prevent the model from forgetting key information of old data. With this design, our model can efficiently and effectively adapt to new coming classes while keeping good performance on already seen data without large-scale retraining. Extensive experiments on the collected dataset demonstrate the favorable performance.