返回研究文献库

物体位姿估计 · 2024 · CVPR · CCF A

SAM-6D: Segment Anything Model Meets Zero-Shot 6D Object Pose Estimation

面向训练未见实例的 RGB-D 零样本 6D pose:以目标 CAD 渲染模板作为物体先验,SAM/FastSAM 产生候选 mask,经 DINOv2 语义、外观和几何评分筛选,再用带 background tokens 的粗到细 3D-3D 对应与加权 SVD 输出所有目标实例的位姿。

研究方向

物体位姿估计

任务层级

Instance-level

物体设定

Unseen objects

输入模态

RGB-Dpoint cloudCAD templates

关键依赖

camera intrinsicstarget CAD modelRGB-D imageCAD-rendered template viewsSAM or FastSAM proposals
完整论文元数据 作者、单位、研究方向、数据集、指标、方法标签和阅读状态 展开
作者
Jiehong Lin, Lihua Liu, Dekun Lu, Kui Jia
作者单位
DexForce Co., Ltd. / The Chinese University of Hong Kong, Shenzhen / South China University of Technology
会议 / 期刊
会议 · CVPR · CCF A
研究方向
物体位姿估计
开源状态
已开源
阅读状态
粗读:已生成 精读:已生成
任务设定
Instance-level · Unseen objects
评估指标
BOP AP BOP AR VSD MSSD MSPD ADD ADD-S ADD(S) / ADD(-S)

论文原始写法:segmentation mAP (IoU 0.50-0.95) / BOP AR (VSD/MSSD/MSPD) / ADD(-S)

关键依赖
camera intrinsicstarget CAD modelRGB-D imageCAD-rendered template viewsSAM or FastSAM proposals

Reading Notes

粗读 / 精读

已生成 src/paper-notes/sam-6d/coarse.md

论文粗读笔记

用于快速判断论文定位、问题设定、核心方法、实验结论和是否值得精读。

论文粗读笔记:SAM-6D: Segment Anything Model Meets Zero-Shot 6D Object Pose Estimation

1. 基本信息

  • 标题:SAM-6D: Segment Anything Model Meets Zero-Shot 6D Object Pose Estimation
  • 作者 / 机构:Jiehong Lin, Lihua Liu, Dekun Lu, Kui Jia;DexForce Co., Ltd.、The Chinese University of Hong Kong, Shenzhen、South China University of Technology
  • 年份 / 会议或期刊:2024 / CVPR
  • 论文链接:arXiv:2311.15707
  • 代码:GitHub
  • 阅读模式:粗读

2. 一句话定位

SAM-6D 是一条面向训练未见目标实例的 RGB-D 零样本 6D pose 管线:给定目标的 CAD 模型、相机内参和杂乱场景 RGB-D,它先让 SAM / FastSAM 生成无类别 mask,再用 DINOv2 与 CAD 渲染模板筛出目标实例,最后通过粗到细 3D-3D 对应估计每个实例的 R,tR,t。

方法输入与输出

项目内容(本站整理)
推理输入一帧场景 RGB-D、相机内参和指定目标的 CAD mesh;由 CAD 离线渲染多视角 RGB-D templates。
最终输出每个发现的目标实例的分割 mask,以及 CAD 物体系到相机系的旋转 RR、平移 tt。
必要条件完整 SAM-6D 自行生成并筛选 masks,不要求用户先给目标 mask;仅运行 PEM 位姿后端时则需 mask。深度、CAD 尺度与相机标定必须一致,zero-shot 指新对象无需专用重训。

2.1 论文图表导读

图表作用粗读建议
Figure 1七个 BOP 核心数据集的输入、目标模型、分割与 pose 定性结果先建立“CAD + RGB-D,输出所有目标实例”的任务画面
Figure 2ISM 与 PEM 的完整流水线最重要的全局图,确认分割在前、点匹配在后
Figure 3PEM 的 coarse / fine point matching精读时再看 background token、SDPT 和 weighted SVD
Table 1novel-object instance segmentation看语义、外观、几何三个评分各自的增益
Table 2BOP 七数据集 pose 主结果必须分开阅读 supervised masks、外部 zero-shot masks 与自带 ISM
Tables 3—6background token、两阶段匹配、SDPT、速度判断核心设计是否值得复杂度与时延
SAM-6D Figure 1: BOP qualitative results from RGB-D input to segmentation and poses
原论文 Figure 1:输入杂乱 RGB 图像、深度图和目标物体 CAD 模型,SAM-6D 在 BOP 七个核心数据集上输出实例 mask 与 6D bounding box。来源:arXiv。图片版权归论文作者及出版方所有,本站仅作论文解读展示。

3. 问题设定

  • 任务:instance-level、unseen-object、zero-shot 6D object pose estimation。测试目标物体不参加该目标的训练或微调,但不是类别级 pose,也不是 CAD-free pose。
  • 输入:一帧杂乱场景 RGB image、depth map、camera intrinsics,以及指定目标物体的 CAD model。
  • 物体先验:方法从 CAD 渲染约 42 个 template views;官方 demo 同样要求 .ply 模型、RGB、深度和内参。
  • 输出:场景中所有目标实例的 instance masks 与相机坐标系下的 rotation RR、translation tt。
  • 不需要:测试时人工 object mask、目标实例 real-image fine-tuning、每个对象的专用 detector。
  • 适用场景:已有正确尺度 CAD 且能采集 RGB-D 的新物体上架、仓储和机器人抓取;目标是从杂乱场景中找出该指定物体并给出姿态。

4. 核心动机

  • 新物体的 6D pose 不能假定有针对该实例训练的 detector 或 pose network;只做 pose、把 segmentation 当作已知输入,会回避实际系统最难的前端问题。
  • SAM 可以给出大量 class-agnostic mask proposal,却不知道“哪个 proposal 是目标 CAD 对应的实例”。只比较全局语义也会被外观相似物、遮挡或背景扰动。
  • 从 mask 回投影得到的场景点云往往不完整、带遮挡和深度噪声;若强迫每个点都匹配 CAD,错误区域会污染刚体变换。

5. 方法速览

SAM-6D Figure 2: ISM and PEM overview
原论文 Figure 2:SAM-6D 由 Instance Segmentation Model(ISM)和 Pose Estimation Model(PEM)组成。ISM 先产生并筛选 mask,PEM 再从粗到细建立 3D-3D correspondence。来源:论文 PDF。图片版权归论文作者及出版方所有,本站仅作论文解读展示。
  1. ISM: Instance Segmentation Model
    • SAM 或 FastSAM 提供所有可能物体的 mask proposals。
    • 对每个 proposal,与 CAD 渲染 templates 比较 DINOv2 特征,并组合三类分数:语义 ssems_{sem}、外观 sappes_{appe}、几何 sgeos_{geo}。
    • 语义分数先粗筛“像不像目标”;patch-level 外观分数补充细节;几何分数用粗 pose 投影的 CAD bounding box 与 mask box 的一致性过滤不合理候选。
  2. PEM: Pose Estimation Model
    • 将有效 mask 的 RGB-D 反投影成 partial scene point cloud,并与 CAD point cloud 建立对应。
    • coarse point matching 先用 196 个点生成并评分 pose hypotheses;fine point matching 在 2048 点上细化。
    • 每一侧加入可学习 background tokens,让遮挡、缺失或错误 mask 对应到背景而非硬匹配到 CAD;最后由 correspondence 的权重做 SVD 恢复 R,tR,t。
SAM-6D Figure 3: pose estimation model with coarse and fine matching
原论文 Figure 3:PEM 上半部分以稀疏点做初始 pose hypothesis,下半部分以 Sparse-to-Dense Point Transformer 将几何上下文传播到稠密点,再用 weighted SVD 得到最终姿态。来源:论文 PDF。图片版权归论文作者及出版方所有,本站仅作论文解读展示。
  • 基础模型 / 技术:SAM 或 FastSAM、DINOv2 ViT-H、ViT feature extraction、Geometric Transformer、Sparse-to-Dense Point Transformer(SDPT)、3D-3D correspondence、weighted SVD。
  • 训练方式:ISM 依赖公开 foundation models 而不为目标对象重新训练;PEM 是在 MegaPose 使用的 ShapeNet Objects 与 Google Scanned Objects 合成训练数据上训练的共享网络。因而“zero-shot”是指没有对测试目标对象再训练,不是完全 training-free。

6. 实验概览

  • 数据集:BOP 七个核心数据集 LM-O、T-LESS、TUD-L、IC-BIN、ITODD、HB、YCB-V。
  • 分割指标:实例分割 mAP,IoU 阈值从 0.50 到 0.95、步长 0.05。
  • pose 指标:BOP AR,为 VSD、MSSD、MSPD 三个 recall 的平均;补充材料还以 ADD(-S) 对比 OVE6D。
SAM-6D Table 1: novel object instance segmentation on BOP datasets
原论文 Table 1:SAM-6D 的 instance segmentation。使用 SAM 并同时加入语义、外观和几何评分时,七个 BOP 数据集平均 mAP 为 48.1;FastSAM 版本为 44.9。来源:论文 PDF。图片版权归论文作者及出版方所有,本站仅作论文解读展示。
SAM-6D Table 2: BOP pose estimation results
原论文 Table 2:BOP pose 主结果。最下行以 SAM-6D 自身的 SAM-based ISM 产生 masks,平均 AR 为 70.4;带十字标记的部分 baseline 使用额外 pose refinement,比较时需同时看输入和 mask 来源。来源:论文 PDF。图片版权归论文作者及出版方所有,本站仅作论文解读展示。
  • 分割:SAM + 三个评分的平均 mAP 为 48.1,高于 CNOS 的 FastSAM 41.2 和 SAM 40.4。只加语义时为 44.0;再加外观、几何后升至 48.1,说明目标筛选不能只靠全局语义。
  • pose:
    • 使用 supervised MaskRCNN masks 时,SAM-6D RGB-D 的平均 AR 为 66.9;这是后端 pose 能力的较干净上界之一。
    • 使用 CNOS FastSAM masks 时,SAM-6D 为 65.3;使用自己的 FastSAM / SAM ISM 时分别为 66.2 / 70.4。
    • 最终 70.4 来自“自带 SAM ISM + RGB-D PEM”,不能与 RGB-only、使用不同 detector、或有 render refinement 的数值直接等价排序。

7. 优缺点

优点

  • 把开放 proposal、目标级筛选和 6D pose 串为一个完整流程,避免默认给 GT mask 的不现实假设。
  • 明确利用 RGB、depth 与 CAD 三类互补线索:语义找候选、模板外观辨识目标、几何和 3D correspondence 定姿。
  • background tokens 对 partial-to-partial 匹配更合适:YCB-V 上它相对 optimal transport 同时提高 AR(81.4 到 84.5)并减少 pose 时间(4.31 s 到 1.36 s)。

缺点 / 风险

  • 不是 CAD-free:目标的正确 CAD、尺寸、内参和 RGB-D 都是部署前提,目标还需渲染多视角 templates。
  • SAM proposal recall 和 DINO matching 是硬前端;目标极小、严重遮挡、与干扰物外观相近时,漏 proposal 或误筛选会直接让后端无从恢复。
  • RGB-D 的深度缺失、透明或镜面物体会同时伤害几何评分和 3D correspondence;论文没有专门的透明 / 反光建模。
  • 速度不是实时级:RTX 3090 上 SAM 版全流程平均 4.37 s / image,FastSAM 版为 1.43 s / image。

8. 对我的启发

  • 可以借鉴的点:将 SAM 视为 class-agnostic proposal generator,而非直接的 target segmenter;用语义、局部外观、投影几何三段式验证 proposal;用 background state 吸收 partial correspondence 中的 unmatched points。
  • 可以作为 baseline 吗:对“有 CAD、RGB-D、训练未见实例、需自主分割”的 BOP / novel-object pose 工作,是强而贴近系统设定的 baseline。对 category-level、CAD-free 或 RGB-only 方法不应直接横比。
  • 是否值得精读:值得。重点是 ssem/sappe/sgeos_{sem}/s_{appe}/s_{geo} 的实现、background token 如何进入双向 soft assignment、以及 SDPT 如何在少量 transformer token 和稠密点间折中。

Related Papers

相关论文