4 个月前

用于行人重识别及更多领域的批处理DropBlock网络

用于行人重识别及更多领域的批处理DropBlock网络

摘要

由于人员再识别任务经常面临姿态变化和遮挡问题,训练卷积神经网络(CNNs)时,一些注意力局部特征往往会被抑制。在本文中,我们提出了一种批处理丢弃块(Batch DropBlock, BDB)网络,该网络由两个分支组成:一个传统的ResNet-50作为全局分支,另一个为特征丢弃分支。全局分支编码全局显著表示。同时,特征丢弃分支包含一个称为批处理丢弃块的注意力特征学习模块,该模块随机丢弃一批输入特征图中的相同区域,以增强局部区域的注意力特征学习。网络随后将来自两个分支的特征进行拼接,提供了一个更加全面且空间分布更广的特征表示。尽管方法简单,我们的方法在人员再识别任务上达到了最先进的性能,并且也适用于一般的度量学习任务。例如,在CUHK03-Detect数据集上,我们实现了76.4%的Rank-1准确率;在Stanford Online Products数据集上,我们获得了83.0%的Recall-1分数,大幅超越了现有工作的表现(超过6%)。

代码仓库

FaizaAslam2424/Faiza-Aslam
pytorch
GitHub 中提及
zjjszj/ps_dm_reid
pytorch
GitHub 中提及
zjjszj/batch-feature-erasing
pytorch
GitHub 中提及
daizuozhuo/batch-dropblock-network
pytorch
GitHub 中提及

基准测试

基准方法指标
person-re-identification-on-cuhk03-labeledBDB (ICCV'19)
MAP: 76.7
Rank-1: 79.4
person-re-identification-on-market-1501-cBDB
Rank-1: 33.79
mAP: 10.95
mINP: 0.32

用 AI 构建 AI

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

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

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
用于行人重识别及更多领域的批处理DropBlock网络 | 论文 | HyperAI超神经