arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2402.18115cs.CVcs.CL
收藏

UniVS:以提示为查询的统一通用视频分割

UniVS: Unified and Universal Video Segmentation with Prompts as Queries

The Hong Kong Polytechnic University(香港理工大学);OPPO Research Institute(OPPO研究院)

Minghan Li, Shuai Li, Xindong Zhang, Lei Zhang

AI总结:

UniVS 提出以提示作为查询的统一视频分割架构,将多种视频分割任务转化为提示引导的目标分割,在 10 个基准上兼顾性能与通用性。

摘要:

尽管统一图像分割(IS)近期取得了进展,开发统一的视频分割(VS)模型仍然是一个挑战。这主要是因为通用的类别指定 VS 任务需要检测所有对象并在连续帧中跟踪它们,而提示引导的 VS 任务则要求在整个视频中利用视觉或文本提示重新识别目标,这使得用同一架构处理不同任务变得困难。我们尝试解决这些问题,并提出了一种新颖的统一 VS 架构,即 UniVS,通过将提示作为查询。UniVS 将来自先前帧的目标提示特征进行平均,作为其初始查询以显式解码掩码,并在掩码解码器中引入了一个目标级提示交叉注意力层,以整合记忆池中的提示特征。通过将先前帧中实体的预测掩码作为它们的视觉提示,UniVS 将不同的 VS 任务转换为提示引导的目标分割,消除了启发式的帧间匹配过程。我们的框架不仅统一了不同的 VS 任务,还自然地实现了通用训练和测试,确保在不同场景下具有稳健的性能。UniVS 在 10 个具有挑战性的 VS 基准上展示了性能与通用性之间值得称道的平衡,涵盖了视频实例、语义、全景、对象和指代分割任务。代码可在 \url{https://github.com/MinghanLi/UniVS} 找到。

英文摘要:

Despite the recent advances in unified image segmentation (IS), developing a unified video segmentation (VS) model remains a challenge. This is mainly because generic category-specified VS tasks need to detect all objects and track them across consecutive frames, while prompt-guided VS tasks require re-identifying the target with visual/text prompts throughout the entire video, making it hard to handle the different tasks with the same architecture. We make an attempt to address these issues and present a novel unified VS architecture, namely UniVS, by using prompts as queries. UniVS averages the prompt features of the target from previous frames as its initial query to explicitly decode masks, and introduces a target-wise prompt cross-attention layer in the mask decoder to integrate prompt features in the memory pool. By taking the predicted masks of entities from previous frames as their visual prompts, UniVS converts different VS tasks into prompt-guided target segmentation, eliminating the heuristic inter-frame matching process. Our framework not only unifies the different VS tasks but also naturally achieves universal training and testing, ensuring robust performance across different scenarios. UniVS shows a commendable balance between performance and universality on 10 challenging VS benchmarks, covering video instance, semantic, panoptic, object, and referring segmentation tasks. Code can be found at \url{https://github.com/MinghanLi/UniVS}.

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

↑