3 个月前

PS-RCNN:通过主体对象抑制检测人群中的次级人类实例

PS-RCNN:通过主体对象抑制检测人群中的次级人类实例

摘要

在高度拥挤场景中检测人体是一项极具挑战性的任务。造成这一难题的两个主要原因在于:1)被严重遮挡的人体实例视觉线索微弱,难以提供足够的信息以实现精准检测;2)这些被严重遮挡的实例更容易在非极大值抑制(Non-Maximum Suppression, NMS)过程中被抑制。为解决上述问题,我们提出了一种两阶段检测器的变体——PS-RCNN。PS-RCNN首先通过一个R-CNN模块(称为P-RCNN)检测那些轻微或未被遮挡的物体;随后,利用人体形状的掩码对已检测到的实例进行抑制,从而使被严重遮挡人体的特征得以凸显。在此基础上,PS-RCNN进一步采用另一个专用于检测严重遮挡人体的R-CNN模块(称为S-RCNN),以补全P-RCNN遗漏的检测目标。最终结果为两个R-CNN模块输出结果的集成。此外,我们引入了高分辨率感兴趣区域对齐(High Resolution RoI Align, HRRA)模块,以尽可能保留严重遮挡人体可见部分的细粒度特征。在CrowdHuman数据集上,与基线方法相比,PS-RCNN分别将召回率(recall)和平均精度(AP)提升了4.49%和2.92%;在WiderPerson数据集上也取得了类似的性能提升。

基准测试

基准方法指标
object-detection-on-crowdhuman-full-bodyPS-RCNN (Faster RCNN, ResNet50, COCO Instance Masks
AP: 87.94
object-detection-on-crowdhuman-full-bodyPS-RCNN (Faster RCNN, ResNet50)
AP: 86.05
object-detection-on-widerpersonPS-RCNN (Faster RCNN, ResNet50)
AP: 89.96

用 AI 构建 AI

从想法到上线——通过免费 AI 协同编程、开箱即用的环境和市场最优价格的 GPU 加速您的 AI 开发

AI 协同编程
即用型 GPU
最优价格
立即开始

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
PS-RCNN:通过主体对象抑制检测人群中的次级人类实例 | 论文 | HyperAI超神经