龙空技术网

把"跟踪一切"搬到无人机上!MIT联合哈佛大学提出开放环境下的实时跟踪模型FAn

将门创投 106

前言:

当前小伙伴们对“ai跟踪算法”都比较关注,我们都需要剖析一些“ai跟踪算法”的相关内容。那么小编同时在网上网罗了一些有关“ai跟踪算法””的相关资讯,希望小伙伴们能喜欢,你们一起来了解一下吧!

目标检测和目标跟踪对于机器人、具身智能智能体而言都非常重要,它们共同构成了机器人系统的基础外部感知单元。近来,以Segment Anything(SAM)为代表的感知一切算法登上了计算机视觉的舞台,并且引发了广泛的关注。

本文介绍一篇MIT和哈佛大学合作完成的研究论文,提出了一种实时检测、跟踪和跟随任何物体的机器人系统,被称为“follow everything”(FAn)与SAM类似,FAn是一种开放词汇和多模态模型,其不限于参与训练的样本类别,可以在推理时根据输入的文本、图像或者以鼠标交互的方式来点击跟踪视频中的目标为了实现这种效果,作者在FAn引入了CLIP、DINO和SAM等基础视觉语言大模型,在输入视频序列中查询和匹配跟踪目标的分割mask和边界框,同时作者还对跟踪过程中受物体遮挡和画面质量影响的目标进行了后处理

下面视频展示了FAn部署在无人机上对地面车辆的跟踪效果,当车辆被其他物体遮挡丢失目标后,FAn可以迅速的实现二次检测和跟踪。考虑到FAn系统的落地实用性,作者对其做了专门的轻量化处理FAn可以在微型无人机(MAV)和轻量级的笔记本显卡(显存为6-8GB)上进行流畅部署,可以达到6-20fps的实时跟踪效果。

文章链接:

代码仓库:

一、引言本文作者认为,现有的跟踪技术仍然存在一些明显的缺点:1)训练的模型只能处理封闭集中的类别,这意味着将这种模型应用在机器人系统上只能处理一组固定的对象类别,这限制了机器人对外界环境的适应性。2)此外,感兴趣的对象只能通过类标签来指定,这对于用户来说是非常不直观的交互方式,尤其是在视频跟踪这种需要进行实时反馈的场景中。目前深度学习社区出现了很多效果拔群的大模型,基于多模态数据预训练的大模型CLIP[1]和DINO[2]等都在开放集场景中表现出了非常惊人的性能,它们不需要再针对特定任务进行训练,就可以在推理时根据用户指定来捕获感兴趣的目标。上图展示了FAn系统在一幅包含4条鲸鱼的输入帧上的运行效果,用户可以选择是在鲸鱼上单击查询还是在整个水面区域单击查询。首先,FAn会调用SAM来提取多个mask,然后根据DINO特征将每个mask分类为它从给定查询(水/鲸鱼)中引用的对象。最后,通过分配DINO特征描述符来检测到与当前查询最接近的鲸鱼。通过以上设计,FAn实现了一个开放式、多模态和高灵活性的实时跟踪一切系统,除了上面例子展示的通过鼠标点击选取感兴趣区域,FAn还提供了通过输入文本提示、图像和边界框等多种交互方式。二、本文方法从FAn整体实现角度来看,其使用了目前最为先进ViT模型的组合,将它们统一到了单个系统中,并专门做了实时优化。如下图所示,FAn使用SAM进行场景目标分割,使用DINO和CLIP进行通用视觉特征提取,对于跟踪模块,作者使用SiamMask[3]框架进行实现。除此之外,作者还设计了一个二次检测机制,可解决对象被遮挡或跟踪丢失的情况,该机制可以自主运行或在人工指导下运行,确保目标被成功识别并再次跟踪,保持跟踪过程的连续性。2.1 开放词汇目标检测和分割 作者首先基于SAM、CLIP和DINO构建了一个轻量级目标检测和分割pipeline,使用视频流的RGB帧 作为输入,查询向量 表示视频中要检测的感兴趣对象(例如,文本“蓝鲸”、鲸鱼的图像,或点击图像中的一条鲸鱼)。作者使用 来表示分割运算符, 以当前帧 为输入,输出一组掩码 ,然后使用 来表示特征提取器模型。 同样接收当前帧 作为输入,并输出描述符张量 FAn首先对每一帧应用 操作来得到二进制的实例分割mask集合:,随后专门计算查询 的特征描述符 ,通过在 上提取像素级描述符得到:其中每个描述符对应于输入图像中的一个像素。为了将每个区域(mask)与给定的输入查询 进行比较,需要聚合这些像素描述符以形成区域级描述符,作者使用了简单的平均池化来实现:其中 表示二进制矩阵 中非零向量的数量,向量 表示包含 区域中所有语义信息的编码,因此,对于每个区域,FAn都拥有其相应的特征描述符 2.2 在边缘计算设备上进行快速检测 由于SAM和DINO等模型的基础版本在处理视频帧时耗时较长,因此其不太适合在实时机载设备上部署,为了解决计算性能瓶颈,作者提出通过对DINO特征进行分组来获得粗略检测结果,这些粗略检测结果可以通过不断迭代计算来进一步细化,从而使整个系统达到高帧速率运行的效果。首先在输入帧 上使用DINO来提取像素级描述符 ,随后计算每个查询 与描述符集合之间的余弦相似度 对于图像中的每个像素,选择与其具有最大余弦相似度的查询 进行聚类。如上图所示,如果查询向量 与像素描述符 之间余弦相似度超过指定阈值 ,就将当前查询的标签分配给原始帧 中的对应像素,否则视为未标记像素。上图展示了使用这种方法检测实际目标的示例,可以看出,虽然这种方法速度更快,但是其产生的分割mask较为粗糙,不够精细。为了进一步加速检测和分割,作者使用quantization和tracing两种模型优化策略对DINO ViT模型进行了处理,其中quantization可以降低模型中数值的精度,从而降低内存需求和计算复杂性。tracing是一种替代优化技术,其可以将动态计算图转换为静态图表示,从而实现模型图计算的高效并行化。上表展示了优化后的DINO以及FAn系统中其他模型的运行时信息,可以看到,在使用quantization和tracing处理后的模型,FPS指标有了明显的提升。2.3 丢失物体二次检测如果由于遮挡或目标突然运动而导致丢失跟踪对象,FAn系统会自动启动二次检测过程,该过程主要通过对目标轨迹进行交叉存储来实现。如下图所示,FAn首先会存储当前跟踪目标的DINO特征,一旦目标丢失,FAn会从跟踪器中获取历史mask,对于每个mask,首先使用与先前相同的方式计算DINO描述符,并且将其与预先计算的描述符进行比较,如果获得高相似度,继续保持跟踪。上述过程的完成,主要得益于作者设计的特征描述符存储机制,具体来说,设 为整数,在跟踪过程中,在每次迭代 处,使 ,定义 表示当前跟踪对象的掩码,首先对当前帧 应用 来获得 ,则可以计算当前跟踪对象的平均描述符为:上述特征代表第 迭代步中的跟踪对象,作者直接将该描述符进行存储并添加到先前计算的描述符集合中来进行更新,整体更新后的特征符存储池如下:

后续过程中,每当FAn系统丢失跟踪对象时,就可以从该存储池中查询一个与当前对象最为接近的特征描述符完成二次检测过程:

三、实验效果

本文实验的硬件系统是一台配备RGB相机的四旋翼飞行器,该飞行器是使用飞行控制软件的Pixhawk进行定制的,飞行器如下图所示。其使用herelink数字传输系统将相机数据与其他遥测数据直接传输到地面计算基站(配备一张NVIDIA GeForce RTX 2070 GPU)。地面基站对接收到的数据运行FAn跟踪算法并通过Mavlink向四旋翼飞行器发送控制命令。对于FAn系统的实时运行性能测试,作者使用Grounded-SAM作为对比baseline方法进行对比实验,下表展示了实验结果,可以看到本文方法在输入帧尺寸在320240和640480两种情况下的性能均优于Grounded-SAM。作者在下图中展示了FAn根据文本查询对低分辨率图像进行自动检测的效果,在图(b)中,由于图像分辨率较低,无人机mask面积太小,导致本文方法错误将查询文本“无人机”匹配为整个背景mask。为了解决整个问题,作者直接删除了面积较大的背景mask,如图(c和d)中,FAn就可以精确地定位到无人机。

为了进一步测试FAn的zero-shot检测效果,作者选取了训练类别集合之外的数据进行测试,尝试使用SAM+CLIP的方式通过文本提示进行实时检测,测试场景分为多种类别,

(1)标准检测,例如“检测鲸鱼”。

(2)基于场景推理的检测,此类任务需要对场景有很好的理解,例如“检测拿着鲸鱼的男孩”。

(3)基于特殊属性的检测,此类任务需要寻找具有特定属性的对象,例如“检测白色的狗”。

(4)基于特殊先验知识的检测,这种任务需要预先了解特定对象,例如其名称/昵称,例如“检测梅西/克里斯蒂亚诺·罗纳尔多”。

(5)基于特殊先验知识和属性的检测,这种场景可以看作是前两种情况的组合,例如“检测皇家马德里球员”。

下图展示了该实验的可视化检测效果。

四、总结本文提出了一种新型的"感知一切"方法,称为FAn系统,其组合了多模态大模型与通用的视觉分割、检测和跟踪方法,并且通过一系列模型推理优化策略,极大的降低了视觉领域的SOTA模型在实时机器人系统上的部署难度。基于多模态大模型的开放词汇感知和多模态信息处理能力,FAn拥有较强的环境适应能力,同时提供了多种灵活的用户交互方式。

参考

[1] A. Radford, J. W. Kim, C. Hallacy, A. Ramesh, G. Goh, S. Agarwal, G. Sastry, A. Askell, P. Mishkin, J. Clark et al., “Learning transferable visual models from natural language supervision,” in International conference on machine learning. PMLR, 2021, pp. 8748–8763.

[2] M. Caron, H. Touvron, I. Misra, H. Jegou, J. Mairal, P. Bojanowski, and A. Joulin, “Emerging properties in self-supervised vision transformers,” in Proceedings of the IEEE/CVF international conference on computer vision, 2021, pp. 9650–9660.

[3] Q. Wang, L. Zhang, L. Bertinetto, W. Hu, and P. H. Torr, “Fast online object tracking and segmentation: A unifying approach,” in Proceedings of the IEEE conference on computer vision and pattern recognition, 2019

作者:seven_

Illustration by IconScout Store from IconScout

-The End-

扫码观看

本周上新!

“AI技术流”原创投稿计划


TechBeat是由将门创投建立的AI学习社区(社区上线480+期talk视频,2400+篇技术干货文章,方向覆盖CV/NLP/ML/Robotis等;每月定期举办顶会及其他线上交流活动,不定期举办技术人线下聚会交流活动。我们正在努力成为AI人才喜爱的高质量、知识型交流平台,希望为AI人才打造更专业的服务和体验,加速并陪伴其成长。


投稿内容

// 最新技术解读/系统性知识分享 //

// 前沿资讯解说/心得经历讲述 //


投稿须知

稿件需要为原创文章,并标明作者信息。

我们会选择部分在深度技术解析及科研心得方向,对用户启发更大的文章,做原创性内容奖励


投稿方式

发送邮件到

chenhongyuan@thejiangmen.com

或添加工作人员微信(chemn493)投稿,沟通投稿详情;还可以关注“将门创投”公众号,后台回复“投稿”二字,获得投稿说明。


>>> 添加小编微信!




关于我“门”

将门是一家以专注于数智核心科技领域新型创投机构,也是北京市标杆型孵化器公司致力于通过连接技术与商业,发掘和培育具有全球影响力的科技创新企业,推动企业创新发展与产业升级。
将门成立于2015年底,创始团队由微软创投在中国的创始团队原班人马构建而成,曾为微软优选和深度孵化了126家创新的技术型创业公司。
如果您是技术领域的初创企业,不仅想获得投资,还希望获得一系列持续性、有价值的投后服务,欢迎发送或者推荐项目给我“门”:


⤵一键送你进入TechBeat快乐星球

标签: #ai跟踪算法