SceneComplete 是一个将多个通用预训练感知模块(VLM、分割、图像修复、Image-to-3D、视觉描述符、位姿估计)串联组合的流水线系统,能够从单张RGB-D图像生成场景中所有可见物体的完整3D网格,包括被大量遮挡的新颖物体,从而为机器人抓取和放置提供精确的三维依据。SceneComplete is a pipeline system that composes multiple general-purpose pre-trained perception modules (VLM, segmentation, image inpainting, Image-to-3D, visual descriptors, pose estimation) in sequence, producing complete 3D meshes of every visible object in the scene from a single RGB-D image, including heavily occluded novel objects, and thereby providing precise 3D grounding for robotic grasping and placement.
机器人在日常杂乱环境中操作,需要对三维场景进行精确理解,才能稳定可靠地抓取和放置物体,并避免碰撞。然而现实中往往只能获得单张RGB-D图像,场景中物体相互遮挡,且多为训练数据之外的新颖物体——这是当前三维场景重建方法面临的核心挑战。Robots operating in everyday cluttered environments need an accurate understanding of the 3D scene in order to grasp and place objects stably and reliably and to avoid collisions. In practice, however, often only a single RGB-D image is available, the objects in the scene occlude one another, and most of them are novel objects outside the training data — this is the core challenge facing current 3D scene reconstruction methods.
"Careful robot manipulation in every-day cluttered environments requires an accurate understanding of the 3D scene, in order to grasp and place objects stably and reliably and to avoid colliding with other objects. In general, we must construct such a 3D interpretation of a complex scene based on limited input, such as a single RGB-D image."
现有方法(如 PartialDecomp、OctMAE、ZeroGrasp)在开放世界场景下存在明显局限:要么仅能预测场景级占据值而无法给出精确物体网格,要么依赖有限的物体类别,难以泛化到真实杂乱环境中的新颖物体。SceneComplete 的核心思路是组合(composing)已有的大型预训练视觉模型,而非端到端训练一个新模型,从而天然具备开放词汇泛化能力,并能随基础模型的改进而不断提升。Existing methods (such as PartialDecomp, OctMAE and ZeroGrasp) have clear limitations in open-world settings: they either predict only scene-level occupancy without yielding precise object meshes, or depend on a limited set of object categories and generalize poorly to novel objects in real cluttered environments. The core idea of SceneComplete is to compose existing large pre-trained vision models rather than train a new model end-to-end, which gives open-vocabulary generalization for free and lets the system keep improving as its foundation models improve.
SceneComplete 流水线由六个顺序模块组成,每个模块均调用独立的预训练大型视觉模型。从单张RGB-D输入出发,逐步完成:物体识别 → 分割 → 图像修复 → Image-to-3D → 尺度估计 → 6D位姿配准,最终输出完整场景的物体网格集合。The SceneComplete pipeline consists of six sequential modules, each invoking an independent pre-trained large vision model. Starting from a single RGB-D input, it proceeds step by step: object recognition → segmentation → image inpainting → Image-to-3D → scale estimation → 6D pose registration, finally outputting the set of object meshes for the complete scene.
输入RGB图像送入 ChatGPT-4o,生成场景中物体的文字描述列表;描述和图像再送入 GroundedSAM2 生成每个物体的分割掩码。采用多策略提示(全局描述 + 部分扩展)并通过 IoU 去重,应对 VLM 漏检和过度分割的情况。The input RGB image is fed to ChatGPT-4o to produce a list of textual descriptions of the objects in the scene; the descriptions and the image are then fed to GroundedSAM2 to generate a segmentation mask for each object. Multi-strategy prompting (global description + partial expansion) combined with IoU-based deduplication handles VLM misses and over-segmentation.
对每个分割出的(可能被遮挡的)物体,用 BrushNet(含 LoRA 适配)进行图像修复,将其转化为白色背景上的单个完整可见物体图像。通过边界框扩展保留上下文信息,适配训练使输出符合后续 Image-to-3D 模型的输入要求。For every segmented (possibly occluded) object, BrushNet (with LoRA adaptation) performs image inpainting, turning it into an image of a single, fully visible object on a white background. Bounding-box expansion preserves contextual information, and adaptation training makes the output match the input requirements of the downstream Image-to-3D model.
将修复后的2D图像送入 InstantMesh,生成带纹理的3D网格。修复步骤对该模块至关重要:未经修复时,Image-to-3D 模型会生成不完整的网格;修复后则能产生准确的3D重建结果(见图4)。The inpainted 2D image is fed to InstantMesh to generate a textured 3D mesh. The inpainting step is critical for this module: without it the Image-to-3D model produces incomplete meshes; with it, accurate 3D reconstructions are obtained (see Figure 4).
利用视觉 Transformer 的密集对应匹配,将预测网格与观测点云对齐,估计各向同性尺度因子。最后用 FoundationPose 进行6自由度位姿估计,将每个网格配准到原始扫描的3D坐标系中,完成场景拼装。Dense correspondence matching from a vision Transformer aligns the predicted mesh with the observed point cloud and estimates an isotropic scale factor. Finally, FoundationPose performs 6-DoF pose estimation, registering every mesh into the 3D coordinate frame of the original scan to assemble the scene.
在 GraspNet-1B 大型基准数据集上进行定量评估(场景重建质量 + 抓取指标),并在 YCB-V 数据集和真实机器人平台上验证抓取成功率。基线包括 PartialDecomp、OctMAE 和 ZeroGrasp。Quantitative evaluation is carried out on the large-scale GraspNet-1B benchmark (scene reconstruction quality + grasp metrics), and grasp success rates are validated on the YCB-V dataset and on a real robot platform. Baselines include PartialDecomp, OctMAE and ZeroGrasp.
| 方法Method | MIoU ↑ | CD ↓ | MMD-EMD ↓ | GC ↓ |
|---|---|---|---|---|
| PartialDecomp | 0.166 | 3.16 | 3.32 | 53.5 |
| OctMAE | 0.445 | 1.73 | 3.11 | 20.3 |
| ZeroGrasp | 0.440 | 1.86 | 3.07 | 18.9 |
| SceneComplete(本文)SceneComplete (ours) | 0.478 | 1.54 | 3.06 | 16.4 |
MIoU: Mean Intersection over Union;CD: Chamfer Distance;MMD-EMD: Modified Maximum Distance using Earth Mover's Distance;GC: Grasp Collision率。SceneComplete 在所有四项指标上均优于所有基线。MIoU: Mean Intersection over Union; CD: Chamfer Distance; MMD-EMD: Modified Maximum Distance using Earth Mover's Distance; GC: Grasp Collision rate. SceneComplete outperforms all baselines on all four metrics.
| 方法Method | Contact-GraspNet GSR | Antipodal GSR | 总体 GSR ↑Overall GSR ↑ |
|---|---|---|---|
| PartialDecomp | 0.46 ± 0.34 | 0.17 ± 0.13 | 0.32 |
| SceneComplete(本文)SceneComplete (ours) | 0.81 ± 0.2 | 0.73 ± 0.18 | 0.77 |
| 方法Method | 成功率 ↑Success rate ↑ |
|---|---|
| Partial Point Cloud(仅部分点云)Partial Point Cloud (partial point cloud only) | 36.7 ± 9.9% |
| OctMAE | 59.6 ± 15.3% |
| SceneComplete(本文)SceneComplete (ours) | 73.3 ± 15.2% |
图4 展示了修复(inpainting)模块的关键作用:去掉修复步骤后,Image-to-3D 模型只能生成残缺网格;未适配的 BrushNet 会引入伪影,而经过 LoRA 适配的版本能正确恢复遮挡区域,生成完整可见物体图像。消融验证了流水线中每个模块(尤其是修复和尺度估计)对最终重建质量的不可替代性。Figure 4 illustrates the critical role of the inpainting module: with the inpainting step removed, the Image-to-3D model can only produce incomplete meshes; the unadapted BrushNet introduces artifacts, whereas the LoRA-adapted version correctly recovers the occluded regions and produces images of fully visible objects. The ablation confirms that every module in the pipeline — inpainting and scale estimation in particular — is indispensable to the final reconstruction quality.
视觉语言模型(VLM)偶尔会遗漏场景中的物体,尤其是外观不典型或被高度遮挡的物体。论文采用多策略提示(multi-strategy prompting)加以缓解,但无法完全消除。The vision-language model (VLM) occasionally misses objects in the scene, especially those with atypical appearance or under heavy occlusion. The paper mitigates this with multi-strategy prompting, but cannot eliminate it entirely.
GroundedSAM 有时会将单个物体的不同部分分开分割。论文通过基于 IoU 的去重(deduplication)部分解决,但仍可能导致下游网格重复或残缺。GroundedSAM sometimes segments different parts of one object separately. The paper partially resolves this through IoU-based deduplication, but it can still lead to duplicated or incomplete meshes downstream.
当前修复方法会移除物体周围的场景上下文信息。通过边界框扩展和模型适配(LoRA)可部分缓解,但对极端遮挡情况效果有限。The current inpainting method removes the scene context surrounding the object. Bounding-box expansion and model adaptation (LoRA) partially mitigate this, but help little under extreme occlusion.
Image-to-3D 模型在处理"highly unusual viewpoints"(极端非常规视角)时会产生失真或不完整的网格,这是当前生成式3D模型的固有局限。The Image-to-3D model produces distorted or incomplete meshes for "highly unusual viewpoints", an inherent limitation of current generative 3D models.
当前方法假设各向同性(isotropic)缩放,对非均匀形状的物体可能不准确。此外,6D位姿配准在"uniformly-textured objects"(纹理均匀的物体)上会失败,因为缺乏可用的区分性特征点。The current method assumes isotropic scaling, which can be inaccurate for non-uniformly shaped objects. In addition, 6D pose registration fails on "uniformly-textured objects", because no usable distinctive feature points are available.
Image-to-3D 和修复模型的输出对随机种子(seed)敏感,相同输入可能产生质量差异较大的结果,影响系统的可重复性。The outputs of the Image-to-3D and inpainting models are sensitive to the random seed; identical inputs can yield results of widely varying quality, which hurts the reproducibility of the system.