首页 > 编程知识 正文

基于深度学习的行人检测,阿里行人重识别算法

时间:2023-05-04 02:30:51 阅读:182062 作者:2548

点击上方“hdsdxh学视觉”,选择添加“星标”或“置顶”重磅干货,第一时间送达作者|叶茫武汉大学

编辑| CV君

自转|最喜欢计算机视觉(微信vml ) ) )。

摘要:行人再识别(Person Re-Identification,简称Re-ID )是一种利用计算机视觉技术来搜索图像或视频序列中是否存在特定行人的AI技术本文介绍了我们最新的IEEE TPAMI综述论文《Deep Learning for Person Re-identification: A Survey and Outlook》,本文作者来自武汉大学、起源人工智能研究院(IIAI )、北理工、英国萨里大学、销售财富亚洲研究院。

首先,感谢为这篇综述提供了非常充实的素材和基础的领域的前辈和大家。

这里是该综述的几个主要贡献点

综述:全面调查了近年来深度学习在Re-ID领域的进展,涵盖了近几年三大视觉峰会上的大部分文章(如有遗漏)。 主要包括Closed-world Re-IDOpen-world Re-ID 的研究进展、常用数据集和评价指标概述,分析了现有方法的不足和改进。

展望(1)用于评估找到与行人匹配最困难的人所需成本的新评估指标mINP); 2 )以强大的AGW 方法,在包含12个数据集的4种Re-ID任务中取得了较好的效果; 3 )从五个不同的方面讨论了未来Re-ID研究的重点和难点,仅供参考。

最新版论文地址: https://arxiv.org/ABS/2001.04193 v2

论文代码和评价指标的开源地址: https://github.com/mangye 16/Reid-survey

3358 www.Sina.com/http://www.Sina.com/http://www.Sina.com /

基于个人理解,本文引用了Re-ID技术的01

数据收集通常来源于每个监视相机的原始视频数据

从行人边框生成、视频数据中,通过人工方式或行人检测或跟踪方式从图中截取行人,图像中的行人占大部分面积;

包含摄像头标签和行人标签等其他信息的训练数据标签

对模型进行再认识和训练,设计模型(主要是深度学习模型),从训练数据中尽可能挖掘“如何识别不同行人的隐性特征表达模式”。

将行人检索和训练的模型应用于测试场景,验证该模型的实际效果。

如下图所示。

图1 .构建行人再识别系统的五个主要步骤。 1 )数据采集;2 )行人边框生成;3 )包括训练数据注释; 4 )再认识模型训练,5 )行人搜索

全文的综述和展望围绕这五个步骤展开和讨论。 关于上述五个步骤的一些约束,本文将ReID技术分为 前言两个大部分集合。 Closed-world是一种完整的通用标记监督行人重识别方法,Open-world概括为多模态数据、端到端行人搜索、无监督或半监督学习、噪声标记和一些Open-set的其他场景。

3358 www.Sina.com/http://www.Sina.com/http://www.Sina.com /

五大步骤

本章主要介绍大家常用的封闭世界(Closed-world ) Re-ID设置。 一般包括以下前提。

用图像、视频、可见光(RGB )照相机捕捉行人

行人从bounding boxes箱出来;

有足够的标记训练数据;

标记的数据标签通常是正确的

query person必须出现在gallery set上。

根据方法设计流程,将其分为特征学习、度量学习和序列优化三部分。 研究者的方法通常对这三个方面进行改进,侧重点不同。 一些提出了新颖的特征学习方法,一些提出了有效的度量损失函数,一些在测试检索阶段进行了优化。 本章最后还总结了现有的常用数据集和评价指标,以及对现有SOTA优缺点的分析。

Closed-world

图2.4种主要的Re-ID特征学习方法

全局特征学习:利用全身全局图像进行特征学习。 常见的改进思路是a

ttention 机制,多尺度融合等;

局部特征学习:利用局部图像区域(行人部件或者简单的垂直区域划分)来进行特征学习,并聚合生成最后的行人特征表示;

辅助特征学习:利用一些辅助信息来增强特征学习的效果,如语义信息(比如行人属性等)、视角信息(行人在图像中呈现的不同方位信息)、域信息(比如每一个摄像头下的数据表示一类域)、GAN 生成的信息(比如生成行人图像)、数据增强等;

视频特征学习:利用一些视频数据提取时序特征,并且融合多帧图像信息来构建行人特征表达

特定的网络设计:利用 Re-ID 任务的特性,设计一些细粒度,多尺度等相关的网络结构,使其更适用于 Re-ID 的场景。

2.2 度量学习方法

图 3. 三种常见的Re-ID度量学习(损失函数)方式

早期的度量学习主要是设计不同类型的距离/相似度度量矩阵。深度学习时代,主要包括不同类型的损失函数的设计及采样策略的改进:

Identity Loss: 将 Re-ID 的训练过程当成图像分类问题,同一个行人的不同图片当成一个类别,常见的有 Softmax 交叉熵损失函数;

Verification Loss:将 Re-ID 的训练当成图像匹配问题,是否属于同一个行人来进行二分类学习,常见的有对比损失函数,二分类损失函数;

Triplet Loss:将 Re-ID 的训练当成图像检索问题,同一个行人图片的特征距离要小于不同行人的特征距离,以及其各种改进;

训练策略的改进:自适应的采样方式(样本不均衡,难易程度)以及不同的权重分配策略

2.3 排序优化

用学习好的 Re-ID 特征得到初始的检索排序结果后,利用图片之间的相似性关系来进行初始的检索结果优化,主要包括重排序(re-ranking)和排序融合(rank fusion)等。

图 4. 重排序示例

2.4 数据集和评价

主要包括现有的一些常用图像和视频数据集的概括,以及现有方法 SOTA 的一些总结和分析,希望综述里面的一些分析能够在大家进行模型设计时提供一些思路和帮助。具体分析详见原文。

      03      

Open-World Re-ID

由于常规的 Closed-world Re-ID 在有监督的实验场景中已经达到或接近瓶颈了,现在很多的研究都更偏向于 Open-World Re-ID 场景,也是当前Re-ID研究的热点。

根据 Re-ID 系统设计的五个步骤,本章节也从以下五个方面介绍:

多模态数据,所采集的数据不是单一的可见光模态;

端到端的行人检索(End-to-end Person Search),没有预先检测或跟踪好的行人图片/视频;

无监督和半监督学习,标注数据有限或者无标注的新场景;

噪声标注的数据,即使有标注,但是数据采集和标注过程中存在噪声或错误;

一些其他 Open-set 场景,查询行人找不到,群体重识别,动态的多摄像头网络等。

3.1 多模态(异构)数据

基于深度图像 Re-ID:旨在利用深度图信息的匹配(融合或跨模态匹配),在很多人机交互的室内场景应用中非常重要;

文本到图像 Re-ID:旨在利用文字语言描述来搜索特定的行人图像,解决实际场景中查询行人图像缺失等问题;

可见光到红外 Re-ID:旨在跨模态匹配白天的可见光图像到夜晚的红外行人图像,也有一些方法直接解决低照度的重识别任务;

跨分辨率 Re-ID:不同高低分辨率行人图像匹配,旨在解决不同距离摄像头下行人分辨率差异巨大等问题

总体而言,异构的行人重识别问题需要解决的一大难题是不同模态数据之间的差异性问题

3.2 端到端Re-ID(End-to-end Person Search)

纯图像/视频的 Re-ID;从原始 raw 图像或者视频中直接检索出行人;

多摄像头跟踪的 Re-ID;跨摄像头跟踪,也是很多产业化应用的重点。

3.3 半监督和无监督的 Re-ID

为了缓解对标注数据的依赖,半监督和无监督/自监督现在成为了当前研究的热点,在 CV 顶会上呈爆炸之势。本文也主要分成两个部分:无监督 Re-ID(不需要标注的源域)和无监督域自适应 Re-ID(需要标注的源域数据或模型):

无监督 Re-ID:主要包括一些跨摄像头标签估计(聚类或图匹配等)的方法,以及一些其他监督信息挖掘的方法(如local patch相似性等);

无监督域自适应 Re-ID:包括一些目标域图像数据生成和一些目标域监督信息挖掘等方式。

考虑到无监督学习也是现在研究的热点,本文也对现有的 SOTA 做了一个简单的总结和分析,可以看到现在的无监督学习方法已经是效果惊人了,未来可期。

如下表所示:

表 1. 现有无监督学习方法在常见数据集上的效果

3.4 噪声鲁棒的Re-ID

主要针对标注数据或者数据采集中产生的一些噪声或错误等,方法包括:

Partial Re-ID:解决行人图像区域部分被遮挡的行人重识别问题;

Noise Sample:主要针对行人图像或视频中检测、跟踪产生的错误或偏差

Noise Label:主要针对行人标签标注产生的错误

3.5 Open-set Re-ID and asjdls

主要针对一些其他开放场景进行一些探讨,如1)gallery set 中 query 行人没有出现的场景;2)Group Re-ID:行人群体匹配的问题;3)动态的多摄像头网络匹配等问题

      04      

展望

4.1 新的评价指标mINP:

考虑到实际场景中,目标人物具有隐匿性,很多时候要找到其困难目标都非常难,给侦查工作带来麻烦。mINP 主要目的是为了衡量 Re-ID 算法用来找到最难匹配样本所要付出的代价,

其计算方式跟 mAP 一样非常简单,应该还是对评价 Re-ID 效果有一定补充作用。简单来讲,排名倒数第一的正确样本位置越靠后,人工排查干预的代价越大,mINP的值越小。这里感谢 Fast-ReID(github.com/JDAI-CV/fast )项目对我们评价指标和方法的集成。

4.2 新的基准方法AGW

主要是在@坚定的大白 的Bag of tricks(感谢)上做的一些改进,主要包括:

Non-local注意力机制的融合;

Generalized-mean (GeM) Pooling的细粒度特征提取;

加权正则化的三元组损失(Weighted Regularization Triplet (WRT) loss)

图 5. AGW 方法的示意图

在审稿人的建议下,我们在四种不同类型的任务(图像Re-ID, 视频Re-ID,跨模态红外Re-ID和Partial Re-ID)的12个数据集上均对我们提出的新指标mINP和AGW方法进行了测评,在大多数情况下,我们的方法都能够取得比较好的效果。具体的实验结果和分析可以参考我们的论文和补充材料,希望我们的方法和评价指标能对大家有一些帮助。

4.3 对未来一些研究方向的思考

这一部分也是紧扣前面提出的五个步骤,针对五个步骤未来亟待解决的关键问题或者热点问题进行归纳。由于每个人理解上的认知偏差,这里的建议仅供大家参考:

不可控的数据采集:不确定多种模态混合的 Re-ID,而不是固定的模态设置;换装的 Re-ID,2020 年已经有好几个新的数据集;

减少人工标注依赖:人机交互的主动学习,选择性的标注;从虚拟数据进行学习(Learning from virtual data),如何解决虚拟数据中的domain gap;

面向 Re-ID 通用网络设计:Domain Generalized Re-ID,如何设计一种在未知场景中也表现优异的模型,如何利用自动化机器学习来设计针对 Re-ID 任务的网络模型;

动态的模型更新:如何以小的代价将学习好的网络模型微调至新摄像头场景中;如何高效的利用新采集的数据(Newly Arriving Data)来更新之前已训练好的模型;

高效的模型部署:轻量型快速的行人重识别算法设计,自适应的针对不同类型的硬件配置(小型的移动手机和大型服务器)调整模型。

后记:很多内容在这里都没办详细描述,具体的细节还请大家移步原文、补充材料以及源代码。因能力有限,部分描述不一定完全客观正确,希望大家能够理解。希望我们的综述能够给领域内的同行提供一些帮助和借鉴。再次感谢各位同行和前辈!欢迎各位批评指正!

下载1:OpenCV-Contrib扩展模块中文版教程

在「hdsdxh学视觉」公众号后台回复:扩展模块中文教程即可下载全网第一份OpenCV扩展模块教程中文版,涵盖扩展模块安装、SFM算法、立体视觉、目标跟踪、生物视觉、超分辨率处理等二十多章内容。

下载2:Python视觉实战项目31讲

在「hdsdxh学视觉」公众号后台回复:Python视觉实战项目31讲即可下载包括图像分割、口罩检测、车道线检测、车辆计数、添加眼线、车牌识别、字符识别、情绪检测、文本内容提取、面部识别等31个视觉实战项目,助力快速学校计算机视觉。

下载3:OpenCV实战项目20讲

在「hdsdxh学视觉」公众号后台回复:OpenCV实战项目20讲即可下载含有20个基于OpenCV实现20个实战项目,实现OpenCV学习进阶。

下载4:leetcode算法开源书

在「hdsdxh学视觉」公众号后台回复:leetcode即可下载。每题都 runtime beats 100% 的开源好书,你值得拥有!

交流群

欢迎加入公众号读者群一起和同行交流,目前有SLAM、三维视觉、传感器、自动驾驶、计算摄影、检测、分割、识别、医学影像、GAN、算法竞赛等微信群(以后会逐渐细分),请扫描下面微信号加群,备注:”昵称+学校/公司+研究方向“,例如:”axdjw + 上海交大 + 视觉SLAM“。请按照格式备注,否则不予通过。添加成功后会根据研究方向邀请进入相关微信群。请勿在群内发送广告,否则会请出群,谢谢理解~

版权声明:该文观点仅代表作者本人。处理文章:请发送邮件至 三1五14八八95#扣扣.com 举报,一经查实,本站将立刻删除。