用SAM模型实现自动驾驶中的无遮挡视频实例分割,无需昂贵标注数据。
Foundation Models for Amodal Video Instance Segmentation in Automated Driving
- 基于SAM模型,通过点提示与记忆追踪实现无遮挡分割。
- 在BDD100K、nuScenes数据集上达到最新最优性能。
- 适合关注自动驾驶感知与少标注学习的研究者。
本文研究自动驾驶场景下的无遮挡视频实例分割任务。以往方法依赖完全标注的视频数据,但此类数据获取困难且成本高,导致分割与跟踪性能难以兼顾。为此,本文提出利用基础模型解决该问题:在微调SAM模型以适应无遮挡实例分割任务的基础上,从可见掩码中采样点作为提示,并通过点记忆存储历史信息。当某实例在后续帧未被检测到时,从记忆中检索其最近点,结合点追踪算法和上一帧的无遮挡掩码,实现跨帧连续分割。该方法在不依赖无遮挡视频标注的前提下,实现了视频级无遮挡实例分割。S-AModal在BDD100K和nuScenes数据集上均取得当前最佳结果。代码已开源。
原文摘要 · Abstract (English)
In this work, we study amodal video instance segmentation for automated driving. Previous works perform amodal video instance segmentation relying on methods trained on entirely labeled video data with techniques borrowed from standard video instance segmentation. Such amodally labeled video data is difficult and expensive to obtain and the resulting methods suffer from a trade-off between instance segmentation and tracking performance. To largely solve this issue, we propose to study the application of foundation models for this task. More precisely, we exploit the extensive knowledge of the Segment Anything Model (SAM), while fine-tuning it to the amodal instance segmentation task. Given an initial video instance segmentation, we sample points from the visible masks to prompt our amodal SAM. We use a point memory to store those points. If a previously observed instance is not predicted in a following frame, we retrieve its most recent points from the point memory and use a point tracking method to follow those points to the current frame, together with the corresponding last amodal instance mask. This way, while basing our method on an amodal instance segmentation, we nevertheless obtain video-level amodal instance segmentation results. Our resulting S-AModal method achieves state-of-the-art results in amodal video instance segmentation while resolving the need for amodal video-based labels. Code for S-AModal is available at https://github.com/ifnspaml/S-AModal.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。