arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2003.08429cs.CVcs.LGeess.IV

STEm-Seg:用于视频实例分割的时空嵌入

STEm-Seg: Spatio-temporal Embeddings for Instance Segmentation in Videos

RWTH Aachen University(亚琛工业大学);Technical University of Munich(慕尼黑工业大学)

Ali Athar, Sabarinath Mahadevan, Aljoša Ošep, Laura Leal-Taixé, Bastian Leibe

AI总结:

本文提出STEm-Seg,一种基于时空嵌入的单阶段、无提议视频实例分割方法,将视频片段视为三维时空体积,通过端到端训练实现分割与跟踪,在多个数据集上达到最先进性能。

摘要:

现有的视频实例分割方法通常采用多阶段流程,遵循基于检测的跟踪范式,并将视频片段建模为一系列图像。这些方法使用多个网络来检测各个帧中的对象,然后在时间上关联这些检测结果。因此,这些方法通常不是端到端可训练的,并且高度针对特定任务。在本文中,我们提出了一种适用于视频实例分割中多种任务的不同方法。具体来说,我们将视频片段建模为单个三维时空体积,并提出了一种新颖的方法,在单阶段中跨空间和时间分割并跟踪实例。我们的问题表述围绕时空嵌入的概念展开,这些嵌入被训练用于在整个视频片段中将属于特定对象实例的像素聚类。为此,我们引入了(i)增强时空嵌入特征表示的新颖混合函数,以及(ii)一个能够推理时间上下文的单阶段、无提议网络。我们的网络以端到端方式训练,以学习时空嵌入以及聚类这些嵌入所需的参数,从而简化推理。我们的方法在多个数据集和任务上取得了最先进的结果。代码和模型可在 https://github.com/sabarim/STEm-Seg 获取。

英文摘要:

Existing methods for instance segmentation in videos typically involve multi-stage pipelines that follow the tracking-by-detection paradigm and model a video clip as a sequence of images. Multiple networks are used to detect objects in individual frames, and then associate these detections over time. Hence, these methods are often non-end-to-end trainable and highly tailored to specific tasks. In this paper, we propose a different approach that is well-suited to a variety of tasks involving instance segmentation in videos. In particular, we model a video clip as a single 3D spatio-temporal volume, and propose a novel approach that segments and tracks instances across space and time in a single stage. Our problem formulation is centered around the idea of spatio-temporal embeddings which are trained to cluster pixels belonging to a specific object instance over an entire video clip. To this end, we introduce (i) novel mixing functions that enhance the feature representation of spatio-temporal embeddings, and (ii) a single-stage, proposal-free network that can reason about temporal context. Our network is trained end-to-end to learn spatio-temporal embeddings as well as parameters required to cluster these embeddings, thus simplifying inference. Our method achieves state-of-the-art results across multiple datasets and tasks. Code and models are available at https://github.com/sabarim/STEm-Seg.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑