给1万帧视频做目标分割,显存占用还不到1.4GB,代码已开源ECCV20

来源:IT之家   发布时间:2022-12-03 17:13:43   阅读量:5218  

咦,好好的藤原百花怎么突然变成了大红版。

给1万帧视频做目标分割,显存占用还不到1.4GB,代码已开源ECCV20

这只紫色的大手,是活着的灭霸吗。一端有带圈可握的短马鞭

如果你以为以上效果只是后期给物体上色,那你就真的被AI骗了这些奇怪的颜色实际上代表了视频对象的分割但是有一点,这个效果真的是一时半会儿分辨不出来

无论是萌妹的飞发:

或者在毛巾和已经改变形状的物体之间来回移动:

AI目标的分割可以说是完美契合,仿佛颜色焊接在一起该方法不仅能高精度分割目标,还能处理超过1万帧的视频而且分段效果始终处于同一水平,视频后半段依然丝滑细腻

更令人惊讶的是,这种方法对GPU的要求很低研究人员表示,在实验过程中,这种方法消耗的GPU内存从未超过1.4GB,要知道,目前类似的基于注意力机制的方法,在普通消费级显卡上甚至无法处理时长超过1分钟的视频

这就是XMem,伊利诺伊大学香槟分校的学者最近提出的一种长视频目标分割方法目前已被ECCV 2022接收,代码开源如此丝滑的效果也吸引了众多网友在Reddit上观看,热度达到800+

网友们开玩笑说:

你为什么把你的手涂成紫色。

谁知道灭霸是否对计算机视觉有兴趣。

模仿人类记忆

目前,已有的视频对象分割方法很多,但要么处理速度慢,对GPU要求高,要么准确率低。

本文提出的方法可以说是兼顾了以上三个方面不仅可以快速分割长视频,帧数也可以达到20FPS,而且在普通的GPU上就可以完成

1968年,心理学家阿特金森和希弗林提出了多重存储模型根据这个模型,人的记忆可以分为三种模式:瞬时记忆,短时记忆和长时记忆

参考上述模型,研究人员还将AI框架分为三种记忆模式。它们是:

及时更新的瞬时记忆

高分辨率工作记忆

长期记忆。

其中,瞬时内存会每帧更新一次,记录画面中的图像信息当工作记忆饱和时,它会被压缩并转移到长时记忆中

当长时记忆饱和后,过时的特征就会伴随着时间被遗忘一般来说,这在处理了几千帧之后就会饱和这样一来,GPU内存就不会因为时间的推移而不足了

一般情况下,视频目标的分割会给出第一帧的图像和目标掩膜,然后模型会跟踪相关的目标,并为后续帧生成相应的掩膜。具体来说,XMem按如下方式处理单个帧:

整个AI框架由三个端到端的卷积网络组成。

查询编码器用于跟踪和提取查询的特定图像特征。

解码器负责获得存储器读取步骤的输出,以生成对象掩码。

值编码器可以组合图像的掩模和目标来提取新的记忆特征值。

最终值编码器提取的特征值将被添加到工作存储器中。

从实验结果来看,该方法在短视频和长视频中都实现了SOTA。

在处理长视频时,XMem的性能并不会伴随着帧数的增加而下降。

研究团队

他毕业于香港科技大学,目前在伊利诺伊大学香槟分校攻读博士学位研究方向是计算机视觉他已经收到了CVPR,NeurIPS,ECCV和其他顶级会议的几篇论文

他现在是伊利诺伊大学香槟分校的助理教授他毕业于苏黎世联邦理工学院,获得博士学位他的研究兴趣是机器学习和计算机视觉

论文地址:

GitHub:

声明:以上内容为本网站转自其它媒体,相关信息仅为传递更多企业信息之目的,不代表本网观点,亦不代表本网站赞同其观点或证实其内容的真实性。投资有风险,需谨慎。

为你推荐

火速护盘!当天制定回购方案,当天下场买股!上市公司真金白银看好后市

火速护盘!当天制定回购方案,当天下场买股!上市公司真金白银看好后市

4月25日,不少上市公司在下跌中果断实施回购。25日晚间,恒瑞医药、建霖家居、劲嘉股份等公司纷纷公告在盘中实施了回购;威...更多

2022-04-26 01:11:00
人弃我取时间已到!券商首席一致看多,A股迎接反攻时刻?

人弃我取时间已到!券商首席一致看多,A股迎接反攻时刻?

4月25日,上证指数重挫逾5%,失守3000点,创下2020年7月以来新低。盘后,券商中国记者采访了多位券商策略、宏观分...更多

2022-04-26 01:11:00
实实在在赚钱了!2021年收入增34%,这家公司要冲击“纯SaaS业务模

实实在在赚钱了!2021年收入增34%,这家公司要冲击“纯SaaS业务模

4月25日,新三板挂牌公司和创科技晒出了2021年“成绩单”,2021年实现收入1.45亿元,同比增长34%;经营现金流...更多

2022-04-26 01:00:00
高玩不买单了水冷头一哥EKWB裁员1/4:业绩持续暴跌

高玩不买单了水冷头一哥EKWB裁员1/4:业绩持续暴跌

虽然AIO一体式水冷很方便,但分体式水冷依然是目前散热最好、效果最炫的方案,只不过价格很贵,只有高玩才会折腾。然而这个高...更多

2022-04-26 00:38:00
IEEE发布首个供应链金融区块链国际标准蚂蚁集团牵头制定

IEEE发布首个供应链金融区块链国际标准蚂蚁集团牵头制定

,IEEE计算机协会区块链和分布式记账标准委员会召开全体会议,《基于区块链的供应链金融标准》正式发布。这是IEEE发布的...更多

2022-04-26 00:37:00
外媒预计在建设P3晶圆厂后三星电子还将在平泽建设P4晶圆厂

外媒预计在建设P3晶圆厂后三星电子还将在平泽建设P4晶圆厂

,据国外媒体报道,此前有报道称,三星电子2020年年中开始在韩国平泽建设的P3晶圆厂,将在下月开始设备的安装,设备的进入...更多

2022-04-26 00:35:00
万华化学首季净利降近两成2021年分红超78亿

万华化学首季净利降近两成2021年分红超78亿

25日盘后,万华化学披露2022年第一季度报告。一季度,公司在营业收入增超30%的情况下,净利润回落近两成。具体来看,一...更多

2022-04-26 00:35:00
盈利下滑、应收账款和现金流状况恶化!泽达易盛股价较高点已跌超八成

盈利下滑、应收账款和现金流状况恶化!泽达易盛股价较高点已跌超八成

董事长、财务总监协助调查,股价暴跌,业绩下滑,笼罩在泽达易盛头上的疑云久久不散。3月17日晚,泽达易盛发布重大事项公告,...更多

2022-04-26 00:34:00