- 专利标题: 一种基于掩码先验和分层聚合Transformer的遮挡行人重识别方法
-
申请号: CN202410406482.4申请日: 2024-04-07
-
公开(公告)号: CN117994823B公开(公告)日: 2024-06-25
- 发明人: 张国庆 , 杨艳 , 张家伟 , 董仕豪 , 郑钰辉
- 申请人: 南京信息工程大学
- 申请人地址: 江苏省南京市江北新区宁六路219号
- 专利权人: 南京信息工程大学
- 当前专利权人: 南京信息工程大学
- 当前专利权人地址: 江苏省南京市江北新区宁六路219号
- 代理机构: 南京苏高专利商标事务所
- 代理商 柏尚春
- 主分类号: G06V40/10
- IPC分类号: G06V40/10 ; G06V10/764 ; G06V10/80 ; G06V10/82 ; G06V10/40 ; G06N3/0455 ; G06N3/0464 ; G06V10/74
摘要:
本发明公开了一种基于掩码先验和分层聚合Transformer的遮挡行人重识别方法,包括以下步骤:(1)获取多种数据集(2)构建掩码语义信息提取器;(3)构建用于处理图片信息,进行提取特征的Transformer骨干网络;(4)构建用于提取细粒度部分特征的分层特征聚合模块;(5)构建用于提取判别性全局特征的多特征补偿模块;(6)合并步骤(2)‑(5),构建基于掩码先验以及分层聚合Transformer的遮挡行人重识别框架;(7)构建用于遮挡行人重识别模型的损失函数;(8)加载由IMAGENET训练的预训练权重,利用Occluded‑Duke数据集的训练集对模型进行训练,测试集对得到的模型进行测试,寻找最优模型;本发明引导模型关注非遮挡身体部位以及提取判别性信息用于行人匹配。
公开/授权文献
- CN117994823A 一种基于掩码先验和分层聚合Transformer的遮挡行人重识别方法 公开/授权日:2024-05-07