SAM-PD: How Far Can SAM Take Us in Tracking and Segmenting Anything in Videos by Prompt Denoising

Tao Zhou, Wenhan Luo, Qi Ye, Zhiguo Shi, Jiming Chen
3/7/2024
cs.CV

No official code found. Best implementation: infZhou/SAM-PD (6 stars, Apache-2.0, last commit 2024-05-20).

Abstract

Recently, promptable segmentation models, such as the Segment Anything Model (SAM), have demonstrated robust zero-shot generalization capabilities on static images. These promptable models exhibit denoising abilities for imprecise prompt inputs, such as imprecise bounding boxes. In this paper, we explore the potential of applying SAM to track and segment objects in videos where we recognize the tracking task as a prompt denoising task. Specifically, we iteratively propagate the bounding box of each object's mask in the preceding frame as the prompt for the next frame. Furthermore, to enhance SAM's denoising capability against position and size variations, we propose a multi-prompt strategy where we provide multiple jittered and scaled box prompts for each object and preserve the mask prediction with the highest semantic similarity to the template mask. We also introduce a point-based refinement stage to handle occlusions and reduce cumulative errors. Without involving tracking modules, our approach demonstrates comparable performance in video object/instance segmentation tasks on three datasets: DAVIS2017, YouTubeVOS2018, and UVO, serving as a concise baseline and endowing SAM-based downstream applications with tracking capabilities.

View on arXivView PDF

Code Implementations(1)

infZhou/SAM-PDCommunity100%
Apache-2.0
60Mar 6, 20242 years agoApache-2.0

Similar Papers

Cite this paper

@article{zhou2024sampd,
  title  = {SAM-PD: How Far Can SAM Take Us in Tracking and Segmenting Anything in Videos by Prompt Denoising},
  author = {Tao Zhou and Wenhan Luo and Qi Ye and Zhiguo Shi and Jiming Chen},
  year   = {2024},
  eprint = {2403.04194},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2403.04194v1}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2403.04194.svg)](https://codeofpaper.com/paper/2403.04194)

Discussion