Count2Density: Crowd Density Estimation without Location-level Annotations

## Контекст Crowd density estimation — это один из важнейших вопросов в области компьютерного зрения, который применяется в различных сферах, включая безопасность, управление трафиком и городской планировку. Несмотря на прогресс в этой области, большинство существующих подходов требуют точных местонахождений объектов в качестве входных данных для обучения моделей. Это сделано необходимым из-за необходимости научить модели выделять и отслеживать отдельных людей. Однако получение таких данных требует значительных усилий и ресурсов, что приводит к затруднениям при масштабировании для реальных сценариев. Многие методы, пытающиеся обойти эту проблему, оказались неэффективными или слишком специфичными для конкретных задач. Наша мотивация заключается в разработке метода, который позволит получить качественные результаты в crowd density estimation без требования мелких местонахождений каждого объекта. ## Метод Мы предлагаем Count2Density — методику, которая позволяет оценивать плотность толпы на основе только счетных данных (то есть, количества людей в области). Наш подход основывается на использовании **Historical Map Bank** (HMB), который хранит прошлые предсказания модели в виде карт плотности. Для получения псевдо-карт плотности, мы используем **Hypergeometric Distribution**, чтобы выбирать случайные точки внутри областей, оцененных как плотные. Кроме того, мы добавили **Self-Supervised Contrastive Spatial Regulariser**, чтобы улучшить способность модели различать области с высокой плотностью от фоновых регионов. Этапы обучения включают начальную инициализацию HMB с помощью **Unsupervised Saliency Estimator** и его последующую итеративную обновление с помощью **Exponential Moving Average (EMA)** предсказанных карт плотности. ## Результаты Мы провели эксперименты на нескольких выборках данных, включая **ShanghaiTech**, **UCF-QNRF** и **JHU-Crowd++**. Наши результаты показали, что Count2Density не только превосходит другие подходы, требующие менее точных данных, но и выигрывает в параметрах F1-score и Mean Absolute Error (MAE) в сравнении с современными лучшими решениями. Эксперименты также продемонстрировали, что каждая составляющая нашего подхода (например, HMB и Spatial Regulariser) имеет существенное влияние на конечные результаты. ## Значимость Наш подход имеет значительный потенциал в реальных приложениях, таких как системы мониторинга толп, системы безопасности и управление трафиком. Он устраняет необходимость в трудоемкой сборке точных данных для обучения, что делает его более удобным и эффективным для реального применения. Мы также показали, что наш подход может быть расширен для задач, требующих оценки плотности в различных условиях, включая движущиеся толпы и различные среды. ## Выводы Мы представили Count2Density — новый подход