Command Palette

Search for a command to run...

3 个月前

用于高效人员搜索的序列化端到端网络

用于高效人员搜索的序列化端到端网络

摘要

行人搜索旨在联合解决行人检测与行人重识别(re-ID)任务。现有方法多基于Faster R-CNN设计端到端网络,然而由于Faster R-CNN采用并行结构,所提取的特征源自区域建议网络(Region Proposal Network)生成的低质量候选框,而非最终检测得到的高质量边界框。由于行人搜索是一项细粒度任务,此类低质量特征会显著降低重识别性能。为解决该问题,本文提出一种顺序式端到端网络(Sequential End-to-end Network, SeqNet),用于提取更优特征。在SeqNet中,检测与重识别被建模为一个渐进式过程,并通过两个子网络依次完成。此外,我们设计了一种鲁棒的上下文二分图匹配算法(Context Bipartite Graph Matching, CBGM),以有效利用上下文信息作为行人匹配的重要补充线索。在两个广泛使用的行人搜索基准数据集CUHK-SYSU和PRW上的大量实验表明,所提方法取得了当前最优的性能。同时,该模型在单张GPU上可达到11.5 fps的推理速度,且能够轻松集成至现有的端到端框架中。

代码仓库

serend1p1ty/SeqNet
官方
pytorch
GitHub 中提及

基准测试

基准方法指标
person-search-on-cuhk-sysuOIM+SeqNet+CBGM
MAP: 94.3
Top-1: 95.0
person-search-on-cuhk-sysuOIM+SeqNet
MAP: 93.4
Top-1: 94.1
person-search-on-cuhk-sysuNAE+SeqNet+CBGM
MAP: 94.8
Top-1: 95.7
person-search-on-cuhk-sysuNAE+SeqNet
MAP: 93.8
Top-1: 94.6
person-search-on-prwNAE+SeqNet+CBGM
Top-1: 87.6
mAP: 47.6
person-search-on-prwOIM+SeqNet
Top-1: 81.7
mAP: 45.8
person-search-on-prwNAE+SeqNet
Top-1: 83.4
mAP: 46.7
person-search-on-prwOIM+SeqNet+CBGM
Top-1: 84.9
mAP: 46.6

用 AI 构建 AI

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

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

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供