arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2007.12831cs.CV
收藏

一种自训练的点监督目标检测与计数方法

A Self-Training Approach for Point-Supervised Object Detection and Counting in Crowds

School of Electrical and Electronics Engineering, Nanyang Technological University(南洋理工大学电气与电子工程学院);Department of Computer Science, City University of Hong Kong(香港城市大学计算机科学系)

Yi Wang, Junhui Hou, Xinyu Hou, Lap-Pui Chau

摘要:

在本文中,我们提出了一种名为Crowd-SDNet的新型自训练方法,该方法使典型的对象检测器仅使用点级注释(即,用点标记对象)来估计拥挤对象的中心点和大小。具体来说,在训练过程中,我们利用可用的点注释来直接监督对象中心点的估计。基于局部均匀分布的假设,我们从点级监督信息初始化伪对象大小,然后利用这些信息通过拥挤感知损失来引导对象大小的回归。同时,我们提出了一个信心和秩序意识的细化方案,不断完善的初始伪对象的大小,使检测器的能力越来越提高,同时检测和计数人群中的对象。此外,针对极端拥挤的场景,我们提出了一种有效的解码方法,以提高检测器的表示能力。在WiderFace基准测试上的实验结果表明,我们的方法在检测和计数任务下都明显优于最先进的点监督方法,即,我们的方法使平均精度提高了10%以上,计数误差减少了31.2%。此外,我们的方法在人群计数和定位数据集上获得了最佳结果(即,ShanghaiTech和NWPU-Crowd)和车辆计数数据集(即,CARPK和PUCPR+)与最先进的检测计数方法进行比较。该代码将在https://github.com/WangyiNTU/Point-supervised-crowd-detection上公开发布。

英文摘要:

In this paper, we propose a novel self-training approach named Crowd-SDNet that enables a typical object detector trained only with point-level annotations (i.e., objects are labeled with points) to estimate both the center points and sizes of crowded objects. Specifically, during training, we utilize the available point annotations to supervise the estimation of the center points of objects directly. Based on a locally-uniform distribution assumption, we initialize pseudo object sizes from the point-level supervisory information, which are then leveraged to guide the regression of object sizes via a crowdedness-aware loss. Meanwhile, we propose a confidence and order-aware refinement scheme to continuously refine the initial pseudo object sizes such that the ability of the detector is increasingly boosted to detect and count objects in crowds simultaneously. Moreover, to address extremely crowded scenes, we propose an effective decoding method to improve the detector's representation ability. Experimental results on the WiderFace benchmark show that our approach significantly outperforms state-of-the-art point-supervised methods under both detection and counting tasks, i.e., our method improves the average precision by more than 10% and reduces the counting error by 31.2%. Besides, our method obtains the best results on the crowd counting and localization datasets (i.e., ShanghaiTech and NWPU-Crowd) and vehicle counting datasets (i.e., CARPK and PUCPR+) compared with state-of-the-art counting-by-detection methods. The code will be publicly available at https://github.com/WangyiNTU/Point-supervised-crowd-detection.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑