用文生图模型的语义理解能力,实现无需标注的物体追踪。
Leveraging Text-to-Image Diffusion Models for Unsupervised Visual Object Tracking

- 将文生图模型的跨注意力机制用于目标定位,通过文本提示激活图像对应区域。
- 在6个数据集上实现优于现有无监督追踪方法的精度,尤其在细粒度语义场景中表现突出。
- 适合对零样本、无标注视频追踪感兴趣的开发者和研究者。
无监督视觉目标追踪是一项挑战性任务,要求在不依赖真实标注的情况下跟踪视频中的任意目标。尽管已有显著进展,现有最先进方法在需要精细理解视频帧中语义与视觉结构信息的场景下仍表现不佳。文生图扩散模型因其能生成准确反映输入提示语义与结构的图像而著称,展现出强大的视觉语义与结构理解能力。基于此,我们从新视角出发,利用预训练文生图扩散模型中编码的丰富语义知识来解决无监督追踪问题。为使原本用于图像生成的扩散模型适应追踪任务,我们将其重新诠释为连接文本与图像模态的桥梁。这一联系通过交叉注意力机制实现:当同时输入文本与图像时,模型会在交叉注意力图中突出显示与文本语义一致的图像区域。因此,我们学习一个代表追踪目标的文本提示,并使其在每帧图像的交叉注意力图中激活对应区域,从而实现基于扩散模型的目标追踪。具体而言,我们的方法Diff-Tracking由两部分组成:初始提示学习器与在线提示更新器。初始提示学习器在首帧生成捕捉目标对象的提示,使扩散模型能够识别目标;在线提示更新器根据运动信息优化提示,确保跨视频帧的一致性追踪。我们在六个具有挑战性的追踪数据集上评估了该方法,验证了其有效性。
原文摘要 · Abstract (English)
Unsupervised visual object tracking is a challenging task that requires following arbitrary targets in videos without training on ground-truth annotations. Despite considerable progress, existing state-of-the-art unsupervised trackers often struggle in scenarios that demand fine-grained understanding of semantic and visual structural information within video frames. Text-to-image diffusion models are well known for their ability to generate images that accurately reflect the semantics and structures described in the input prompt, demonstrating a strong grasp of visual semantics and structures. Building on this capability, we approach the unsupervised tracking from a new perspective by exploiting the rich semantic knowledge encoded in pretrained text-to-image diffusion models. To adapt the diffusion models, which are originally developed for image generation, to the tracking task, we reinterpret the models as a bridge between text and image modalities. This connection is realized through the cross-attention mechanism: when both text and an image are input into the models, they highlight the regions of the image that are semantically aligned with the text in the cross-attention maps. We therefore learn a prompt that represents the tracking target and activates its corresponding region in the cross-attention map for each frame, which enables object tracking with the diffusion model. Specifically, our method Diff-Tracking is composed of two main components: an initial prompt learner and an online prompt updater. The initial prompt learner generates a prompt that captures the target object in the first frame, allowing the diffusion model to identify the target. The online prompt updater refines the prompt based on motion information, enabling consistent tracking across video frames. We evaluate our approach on six challenging tracking datasets demonstrate the effectiveness of our approach.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。