数ショットカウントは、同じクラスの少数のエグザンプラのみが与えられた場合に、画像内の任意のクラスのオブジェクトをカウントすることを目的としています。既存の相関ベースの数ショットカウントアプローチは、相関の粗さと意味レベルの低さに悩まされています。これらの問題を解決するために、画像とエグザンプラ間の相関に基づいてエグザンプラ関連の機能を段階的に改良する反復フレームワークを提案します。次に、密度マップは、最終的な洗練された特徴マップから予測されます。反復フレームワークには、相関蒸留モジュールと機能改良モジュールが含まれます。反復中に、エグザンプラ関連の機能は徐々に洗練され、エグザンプラに関連しない機能は抑制され、エグザンプラ関連の機能がより重要である場合の数ショットカウントに役立ちます。私たちのアプローチは、数ショットカウントベンチマークFSC-147ですべてのベースラインを大幅に上回っています。驚いたことに、一般的なクラスにとらわれないカウント用に設計されていますが、私たちのアプローチは、車のカウントベンチマークCARPKとPUCPR +、および群集カウントのベンチマークUCSDとMallで最先端のパフォーマンスを実現しています。また、群衆カウントベンチマークShanghaiTechで競争力のあるパフォーマンスを達成します。コードはまもなくリリースされます。
Few-shot counting aims to count objects of any class in an image given only a few exemplars of the same class. Existing correlation-based few-shot counting approaches suffer from the coarseness and low semantic level of the correlation. To solve these problems, we propose an iterative framework to progressively refine the exemplar-related features based on the correlation between the image and exemplars. Then the density map is predicted from the final refined feature map. The iterative framework includes a Correlation Distillation module and a Feature Refinement module. During the iterations, the exemplar-related features are gradually refined, while the exemplar-unrelated features are suppressed, benefiting few-shot counting where the exemplar-related features are more important. Our approach surpasses all baselines significantly on few-shot counting benchmark FSC-147. Surprisingly, though designed for general class-agnostic counting, our approach still achieves state-of-the-art performance on car counting benchmarks CARPK and PUCPR+, and crowd counting benchmarks UCSD and Mall. We also achieve competitive performance on crowd counting benchmark ShanghaiTech. The code will be released soon.