upcount

Overview

Model overview: The model uses 384 x 384 training crops, a class-token-free ViT-B/16, intermediate blocks [2, 5, 8, 11], and a 64-channel stride-four refined feature map. FeatUp-style JBU restores image-aligned detail before the proposal and verification heads.

Training follows the CounTR-length two-stage FSC-147 schedule, followed by optional CARPK fine-tuning. Learning rates are linearly scaled as base_lr * effective_batch_size / 256, with AdamW and cosine decay.

Full paper is available on arXiv, and code available on GitHub.

Citation

@article{wijaya2026spatially,
  title={Spatially-Aware Class-Agnostic Object Counting},
  author={Wijaya, Robert and Hossain, Md Tanvir and Kau, Amanda and Cheung, Ngai-Man},
  journal={arXiv preprint arXiv:2607.16826},
  year={2026}
}