新数据集解决人群头检测难题,提升高危场景安全监控能力
RPEE-HEADS: A Novel Benchmark for Pedestrian Head Detection in Crowd Videos
- 构建高分辨率人群头部数据集,含109,913个标注头像
- 实时检测模型达90.8%精度,推理仅需14毫秒
- 专为铁路站台和活动入口设计,适合安防应用
在铁路站台和活动入口等高风险场景中,自动检测密集人群中的行人头部对人群分析与管理至关重要。现有公开数据集对此类场景覆盖不足,导致深度学习模型性能受限。为此,本文提出RPEE-Heads数据集,包含66段视频中的1,886张图像,共标注109,913个行人头部,平均每图56.2个,标注内容为可见头部的边界框。本文还评估了八种先进目标检测算法在该数据集上的表现,并分析头部大小对检测精度的影响。实验结果表明,You Only Look Once v9和Real-Time Detection Transformer表现最优,平均精度分别达到90.7%和90.8%,推理时间分别为11毫秒和14毫秒。研究强调了专用数据集在提升特定场景头部检测准确性方面的重要性。数据集与预训练模型已开源。
原文摘要 · Abstract (English)
The automatic detection of pedestrian heads in crowded environments is essential for crowd analysis and management tasks, particularly in high-risk settings such as railway platforms and event entrances. These environments, characterized by dense crowds and dynamic movements, are underrepresented in public datasets, posing challenges for existing deep learning models. To address this gap, we introduce the Railway Platforms and Event Entrances-Heads (RPEE-Heads) dataset, a novel, diverse, high-resolution, and accurately annotated resource. It includes 109,913 annotated pedestrian heads across 1,886 images from 66 video recordings, with an average of 56.2 heads per image. Annotations include bounding boxes for visible head regions. In addition to introducing the RPEE-Heads dataset, this paper evaluates eight state-of-the-art object detection algorithms using the RPEE-Heads dataset and analyzes the impact of head size on detection accuracy. The experimental results show that You Only Look Once v9 and Real-Time Detection Transformer outperform the other algorithms, achieving mean average precisions of 90.7% and 90.8%, with inference times of 11 and 14 milliseconds, respectively. Moreover, the findings underscore the need for specialized datasets like RPEE-Heads for training and evaluating accurate models for head detection in railway platforms and event entrances. The dataset and pretrained models are available at https://doi.org/10.34735/ped.2024.2.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。