Unified ViT-CNN for few-shot object counting
摘要
Few-shot object counting aims to count an object of an arbitrary category using only a few annotated exemplars, i.e., few-shot. Existing methods have shown promising results in few-shot counting by using the transformer model combined with convolutions. However, these methods’ approaches to extracting the query and exemplars features using the separated feature extractors, namely Vision Transformer (ViT) and Convolution Neural Network (CNN), respectively, can cause the inconsistency of latent spaces. To address this issue, in this work, we proposed the ViT-CNN architecture sharing feature extraction for query and exemplar extraction by adapting the ViT-Adapter model as the backbone. Our proposed architecture also takes advantage of multi-scale features and the exemplar iterative interaction to enhance the result. Furthermore, we proposed a Density Map Refinement Module to refine the predicted density map using the exemplar information. Extensive experiments on the FSC-147 and CARPK datasets show that our approach significantly outperforms previous state-of-the-art approaches (7.20% error reduction w.r.t. test MAE and 6.35% w.r.t. test RMSE).