RobotSeg:面向图像与视频中机器人分割的模型与数据集
RobotSeg: A Model and Dataset for Segmenting Robots in Image and Video
把 SAM 2 改造成面向机器人本体的自动视频分割器:结构增强的记忆关联器处理关节形变,机器人提示词生成器免去人工提示,标签高效训练策略降低逐帧掩码标注需求,可直接迁移到任意需要免提示视频分割的类别。
原文依据 RobotSeg is built upon the versatile SAM 2 foundation model but addresses its three limitations for robot segmentation
摘要译文
由模型翻译 · 术语统一 · 请对照原文核对关键结论准确的机器人分割是机器人感知的一项基础能力。它可支撑 VLA 系统的精确视觉伺服、可扩展的以机器人为中心的数据增强、准确的真实到仿真迁移,以及在动态人机环境中的可靠安全监控。尽管现代分割模型能力很强,但令人意外的是,分割机器人仍然颇具挑战。这源于机器人本体的多样性、外观的歧义性、结构的复杂性以及形状的快速变化。为应对这些挑战,我们提出 RobotSeg,一个用于图像与视频中机器人分割的基础模型。RobotSeg 构建于通用的 SAM 2 基础模型之上,但针对其在机器人分割上的三点局限——缺乏对关节型机器人的适应、依赖人工提示词、以及需要逐帧训练掩码标注——分别引入结构增强的记忆关联器、机器人提示词生成器和标签高效训练策略加以解决。这些创新共同实现了结构感知、自动且标签高效的解决方案。我们进一步构建了视频机器人分割(VRS)数据集,包含超过 2.8k 段视频(138k 帧),涵盖多样的机器人本体与环境。大量实验表明,RobotSeg 在图像和视频上均达到最优性能,为机器人感知的未来进展奠定了坚实基础。
英文原文摘要 1433 字符
Accurate robot segmentation is a fundamental capability for robotic perception. It enables precise visual servoing for VLA systems, scalable robot-centric data augmentation, accurate real-to-sim transfer, and reliable safety monitoring in dynamic human-robot environments. Despite the strong capabilities of modern segmentation models, surprisingly it remains challenging to segment robots. This is due to robot embodiment diversity, appearance ambiguity, structural complexity, and rapid shape changes. Embracing these challenges, we introduce RobotSeg, a foundation model for robot segmentation in image and video. RobotSeg is built upon the versatile SAM 2 foundation model but addresses its three limitations for robot segmentation, namely the lack of adaptation to articulated robots, reliance on manual prompts, and the need for per-frame training mask annotations, by introducing a structure-enhanced memory associator, a robot prompt generator, and a label-efficient training strategy. These innovations collectively enable a structure-aware, automatic, and label-efficient solution. We further construct the video robot segmentation (VRS) dataset comprising over 2.8k videos (138k frames) with diverse robot embodiments and environments. Extensive experiments demonstrate that RobotSeg achieves state-of-the-art performance on both images and videos, establishing a strong foundation for future advances in robot perception.