改进视频分割模型,让物体追踪更准更稳。
MoSAM: Motion-Guided Segment Anything Model with Spatial-Temporal Memory Selection
- 用运动信息引导提示,提升追踪方向准确性
- 动态筛选可靠历史帧,避免错误记忆干扰
- 适合需要长时稳定分割的视频任务
最近的 Segment Anything Model 2 (SAM2) 在图像和视频的交互式对象分割上表现出色。然而,作为基础交互分割模型,SAM2 仅依赖前六帧的掩码记忆进行分割,导致两个显著问题:一是在视频推理中,因缺乏运动信息,物体可能因消失而无法持续追踪,限制了长距离追踪能力;二是记忆来自固定过去帧,当物体遮挡或消失时,可能导致记忆中存在不准确的分割结果。为此,我们提出 MoSAM,引入两项关键策略:一是提出运动引导提示(MGP),以稀疏与稠密方式表征物体运动,并通过运动引导提示注入 SAM2,使模型能关注运动方向,增强追踪能力;二是设计时空记忆选择(ST-MS)机制,动态识别在像素级与帧级上可能包含准确分割的帧,剔除潜在错误的掩码预测,从而利用更可靠的特征记忆,提升相似区域的分割效果。在多个视频对象分割与视频实例分割基准测试中,MoSAM 的表现优于现有方法,达到当前最优水平。
原文摘要 · Abstract (English)
The recent Segment Anything Model 2 (SAM2) has demonstrated exceptional capabilities in interactive object segmentation for both images and videos. However, as a foundational model on interactive segmentation, SAM2 performs segmentation directly based on mask memory from the past six frames, leading to two significant challenges. Firstly, during inference in videos, objects may disappear since SAM2 relies solely on memory without accounting for object motion information, which limits its long-range object tracking capabilities. Secondly, its memory is constructed from fixed past frames, making it susceptible to challenges associated with object disappearance or occlusion, due to potentially inaccurate segmentation results in memory. To address these problems, we present MoSAM, incorporating two key strategies to integrate object motion cues into the model and establish more reliable feature memory. Firstly, we propose Motion-Guided Prompting (MGP), which represents the object motion in both sparse and dense manners, then injects them into SAM2 through a set of motion-guided prompts. MGP enables the model to adjust its focus towards the direction of motion, thereby enhancing the object tracking capabilities. Furthermore, acknowledging that past segmentation results may be inaccurate, we devise a Spatial-Temporal Memory Selection (ST-MS) mechanism that dynamically identifies frames likely to contain accurate segmentation in both pixel- and frame-level. By eliminating potentially inaccurate mask predictions from memory, we can leverage more reliable memory features to exploit similar regions for improving segmentation results. Extensive experiments on various benchmarks of video object segmentation and video instance segmentation demonstrate that our MoSAM achieves state-of-the-art results compared to other competitors.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。