Saved in:
Bibliographic Details
Main Authors: Pan, Tianhang, Jia, Xiuyi
Format: Preprint
Published: 2024
Subjects:
Online Access:https://arxiv.org/abs/2408.05956
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866908353302626304
author Pan, Tianhang
Jia, Xiuyi
author_facet Pan, Tianhang
Jia, Xiuyi
contents Currently, most crowd counting methods have outstanding performance under normal weather conditions. However, our experimental validation reveals two key obstacles limiting the accuracy improvement of crowd counting models: 1) the domain gap between the adverse weather and the normal weather images; 2) the weather class imbalance in the training set. To address the problems, we propose a two-stage crowd counting method named Multi-queue Contrastive Learning (MQCL). Specifically, in the first stage, our target is to equip the backbone network with weather-awareness capabilities. In this process, a contrastive learning method named multi-queue MoCo designed by us is employed to enable representation learning under weather class imbalance. After the first stage is completed, the backbone model is "mature" enough to extract weather-related representations. On this basis, we proceed to the second stage, in which we propose to refine the representations under the guidance of contrastive learning, enabling the conversion of the weather-aware representations to the normal weather domain. Through such representation and conversion, the model achieves robust counting performance under both normal and adverse weather conditions. Extensive experimental results show that, compared to the baseline, MQCL reduces the counting error under adverse weather conditions by 22%, while introducing only about 13% increase in computational burden, which achieves state-of-the-art performance.
format Preprint
id arxiv_https___arxiv_org_abs_2408_05956
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Boosting Adverse Weather Crowd Counting via Multi-queue Contrastive Learning
Pan, Tianhang
Jia, Xiuyi
Computer Vision and Pattern Recognition
Currently, most crowd counting methods have outstanding performance under normal weather conditions. However, our experimental validation reveals two key obstacles limiting the accuracy improvement of crowd counting models: 1) the domain gap between the adverse weather and the normal weather images; 2) the weather class imbalance in the training set. To address the problems, we propose a two-stage crowd counting method named Multi-queue Contrastive Learning (MQCL). Specifically, in the first stage, our target is to equip the backbone network with weather-awareness capabilities. In this process, a contrastive learning method named multi-queue MoCo designed by us is employed to enable representation learning under weather class imbalance. After the first stage is completed, the backbone model is "mature" enough to extract weather-related representations. On this basis, we proceed to the second stage, in which we propose to refine the representations under the guidance of contrastive learning, enabling the conversion of the weather-aware representations to the normal weather domain. Through such representation and conversion, the model achieves robust counting performance under both normal and adverse weather conditions. Extensive experimental results show that, compared to the baseline, MQCL reduces the counting error under adverse weather conditions by 22%, while introducing only about 13% increase in computational burden, which achieves state-of-the-art performance.
title Boosting Adverse Weather Crowd Counting via Multi-queue Contrastive Learning
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2408.05956