PVT-Crowd: Bridging Multi-scale Features from Pyramid Vision Transformer for Weakly-Supervised Crowd Counting
摘要
Weakly-supervised crowd counting does not require location-level annotations, but only relies on count-level annotations to achieve the task of crowd counting for images, which is becoming a new research hotspot in the field of crowd counting. Currently, weakly-supervised crowd counting networks based on deep learning mostly use Transformers to extract features and establish global contexts, ignoring feature information at different scales, resulting in insufficient feature utilization. In this paper, we propose a well-designed end-to-end crowd counting network named PVT-Crowd bridging multi-scale features from the Pyramid Visual Transformer Encoder for weakly-supervised crowd counting. Specifically, Adjacent-Scale Bridging Modules (ASBM) enable the interaction of high-scale semantic and low-scale detailed information from both channel and spatial dimensions. The Global-Scale Bridging Module (GSBM) performs a secondary fusion of multi-scale feature information. Extensive experiments show that our PVT-Crowd outperforms most weakly-supervised crowd counting networks and obtains competitive performance compared to fully-supervised ones. In particular, cross-dataset experiments confirm that our PVT-Crowd had a remarkable generality.