「AI透视眼」,三次马尔奖获得者Andrew带队解决恣意物体遮拦补全难题

遮拦是计算机视觉很基础但依旧未解决的问题之一,因为遮拦意味着视觉信息的缺失,而机器视觉系统却依靠着视觉信息进行感知和理解,并且在现实世界中,物体之间的相互遮拦无处不在。牛津大学 VGG 实验室 Andrew Zisserman 团队最新事情系统性解决了恣意物体的遮拦补全问题,并且为这一问题提出了一个新的更加精确的评价数据集。该事情受到了 MPI 大佬 Michael Black、CVPR 官方账号、南加州大学计算机系官方账号等在 X 平台的点赞。以下为论文「Amodal Ground Truth and Compl

遮拦是计算机视觉很基础但依旧未解决的问题之一,因为遮拦意味着视觉信息的缺失,而机器视觉系统却依靠着视觉信息进行感知和理解,并且在现实世界中,物体之间的相互遮拦无处不在。牛津大学 VGG 实验室 Andrew Zisserman 团队最新事情系统性解决了恣意物体的遮拦补全问题,并且为这一问题提出了一个新的更加精确的评价数据集。该事情受到了 MPI 大佬 Michael Black、CVPR 官方账号、南加州大学计算机系官方账号等在 X 平台的点赞。以下为论文「Amodal Ground Truth and Completion in the Wild」的主要内容。

「AI透视眼」,三次马尔奖获得者Andrew带队解决恣意物体遮拦补全难题

论文链接:https://arxiv.org/pdf/2312.17247.pdf

项目主页:https://www.robots.ox.ac.uk/~vgg/research/amodal/

代码地址:https://github.com/Championchess/Amodal-Completion-in-the-Wild

非模态分隔(Amodal Segmentation)旨在补全物体被遮拦的部分,即给出物体可见部分和不可见部分的形态掩码。这个义务可以使得诸多下游义务受益:物体识别、目标检测、实例分隔、图像编辑、三维重建、视频物体分隔、物体间支撑关系推理、机器人的操纵和导航,因为在这些义务中知道被遮拦物体残缺的形态会有所帮助。

「AI透视眼」,三次马尔奖获得者Andrew带队解决恣意物体遮拦补全难题

然而,如何去评价一个模型在实在世界做非模态分隔的性能却是一个难题:虽然很多图片中都有大量的被遮拦物体,可是如何失去这些物体残缺形态的参考标准 或是非模态掩码呢?前人的事情有通过人手动标注非模态掩码的,可是这样标注的参考标准难以避免引入人类误差;也有事情通过制造合成数据集,比如在一个残缺的物体上贴直接另一个物体,来失去被遮拦物体的残缺形态,但这样失去的图片都不是实在图片场景。因此,这个事情提出了通过 3D 模型投影的方法,构造了一个大规模的涵盖多物体种类并且提供非模态掩码的实在图片数据集(MP3D-Amodal)来精确评价非模态分隔的性能。各差别数据集的对比如下图:

「AI透视眼」,三次马尔奖获得者Andrew带队解决恣意物体遮拦补全难题

具体而言,以 MatterPort3D 数据集为例,对于恣意的有实在照片并且有场景三维结构的数据集,我们可以将场景中所有物体的三维形态同时投影到相机上以失去每个物体的模态掩码(可见形态,因为物体相互之间有遮拦),然后将场景中每个物体的三维形态分别投影到相机以失去这个物体的非模态掩码,即残缺的形态。通过对比模态掩码和非模态掩码,即可以挑选出被遮拦的物体。

「AI透视眼」,三次马尔奖获得者Andrew带队解决恣意物体遮拦补全难题

数据集的统计数据如下:

「AI透视眼」,三次马尔奖获得者Andrew带队解决恣意物体遮拦补全难题

「AI透视眼」,三次马尔奖获得者Andrew带队解决恣意物体遮拦补全难题

数据集的样例如下:

「AI透视眼」,三次马尔奖获得者Andrew带队解决恣意物体遮拦补全难题

此外,为解决恣意物体的残缺形态重建义务,作者提取出 Stable Diffusion 模型的特征中关于物体残缺形态的先验知识,来对恣意被遮拦物体做非模态分隔,具体的架构如下(SDAmodal):

「AI透视眼」,三次马尔奖获得者Andrew带队解决恣意物体遮拦补全难题

提出使用 Stable Diffusion Feature 的动机在于,Stable Diffusion 具有图片补全的能力,所以可能一定程度上包含了有关物体的全部信息;而且由于 Stable Diffusion 经过大量图片的训练,所以可以期待其特征在恣意环境下有对恣意物体的处理能力。和前人 two-stage 的框架差别,SDAmodal 不需要已经标注好的遮拦物掩码作为输入;SDAmodal 架构简单,却体现出很强的零样本泛化能力(比较下表 Setting F 和 H,仅在 COCOA 上训练,却能在另一个差别域、差别类别的数据集上有所提升);即使没有关于遮拦物的标注,SDAmodal 在目前已有的涵盖多种类被遮拦物体的数据集 COCOA 以及新提出的 MP3D-Amodal 数据集上,都取得了SOTA表现(Setting H)。

「AI透视眼」,三次马尔奖获得者Andrew带队解决恣意物体遮拦补全难题

除了定量实验,定性的比较也体现出了 SDAmodal 模型的优势:从下图可以观察到(所有模型都只在 COCOA 上训练),对于差别种类的被遮拦物体,无论是来自于 COCOA,还是来自于另一个MP3D-Amodal,SDAmodal 都能大大提升非模态分隔的效果,所预测的非模态掩码更加接近实在的。

「AI透视眼」,三次马尔奖获得者Andrew带队解决恣意物体遮拦补全难题

更多细节,请阅读论文原文。

给TA打赏
共{{data.count}}人
人已打赏
工程

ICLR 2024 Spotlight | 大言语模型权重、激活的全方位低bit可微量化,已集成进商用APP

2024-3-7 14:46:00

工程

清华姚班本科生连发两作,十年来最大改善:矩阵乘法接近理论最优

2024-3-8 15:12:00

0 条回复 A文章作者 M管理员
    暂无讨论,说说你的看法吧
个人中心
今日签到
搜索