Abstract
We propose an end-to-end one-step person search approach with learnable proposals, named LEAPS. Given a set of sparse and learnable proposals, LEAPS employs a dynamic person search head to directly perform person detection and corresponding re-id feature generation without non-maximum suppression post-processing. The dynamic person search head comprises a detection head and a novel flexible re-id head. Our flexible re-id head first employs a dynamic region-of-interest (RoI) operation to extract discriminative RoI features of the proposals. Then, it generates re-id features using a plain and a hierarchical interaction re-id module. To better guide discriminative re-id feature learning, we introduce a diverse re-id sample matching strategy, instead of bipartite matching in detection head. Comprehensive experiments reveal the benefit of the proposed LEAPS, achieving a favorable performance on two public person search benchmarks: CUHK-SYSU and PRW. When using the same ResNet50 backbone, our LEAPS obtains a mAP score of 55.0%, outperforming the best reported results in literature by 1.7%, while achieving around a two-fold speedup on the challenging PRW dataset. Our source code and models will be released.
Abstract (translated)
我们提出了一种具有可学习提案的全方位一步人搜索方法,名为LEAPS。给定一组稀疏且可学习的提案,LEAPS使用动态人搜索头直接进行人检测和对应的重写识别特征生成,而不需要非最大抑制后处理。动态人搜索头由检测头和创新的 flexible re-id头组成。我们的 flexible re-id头首先使用动态区域兴趣(RoI)操作提取提案中的区域兴趣特征。然后,它使用一个 plain 和Hierarchical interaction re-id module 生成重写识别特征。为了更好地指导重写识别特征的学习,我们引入了多种重写样本匹配策略,而不是在检测头中使用二元匹配。全面实验揭示了 proposed LEAPS 的优点,在两个公共人搜索基准上取得了有利的性能:CUHK-SYSU 和 PRW。使用相同的 ResNet50 骨干,我们的 LEAPS 获得 55.0% 的 mAP 得分,比文献中报道的最佳结果高出 1.7%,并在挑战性的PRW数据集上实现了大约两倍的速度提升。我们的源代码和模型将发布。
URL
https://arxiv.org/abs/2303.11859