Source-linked AI summary

MeshSegmenter: Zero-Shot Mesh Semantic Segmentation via Texture Synthesis

Ziming Zhong, Yanxu Xu, Jing Li, Jiale Xu, Zhengxin Li, Chaohui Yu, Shenghua Gao

arXiv:2407.13675v3cs.CV

TL;DR

Limited annotated 3D data and ambiguity in local mesh regions make zero-shot 3D segmentation challenging. MeshSegmenter combines generated textures, 2D segmentation of rendered views, and multi-view confidence revoting. The authors report accurate, stable segmentation across diverse meshes and text descriptions.

  • Problem

    Limited annotated 3D data and ambiguity in local mesh regions challenge generalization and accurate text-guided 3D segmentation.

    Source: §1 · p. 2
  • Method

    MeshSegmenter synthesizes textures, segments rendered textured and untextured views with 2D models, and revotes confidence across views on the mesh.

    Source: §1 · p. 3
  • Results

    The authors report precise segmentation across diverse meshes and segment descriptions, with multi-view revoting supporting stable, consistent results.

    Source: §1 · p. 3
  • Takeaways & Limitations

    The framework demonstrates text-guided segmentation of geometrically non-prominent regions, including car doors, and supports mesh editing applications.

    Source: §1 · pp. 2–3, §6 · p. 14

    Visibility of faces remains an issue to consider when inferring 3D segmentation from multi-view 2D detections.

    Source: §5 · p. 14

Abstract

from arXiv · show

We present MeshSegmenter, a simple yet effective framework designed for zero-shot 3D semantic segmentation. This model successfully extends the powerful capabilities of 2D segmentation models to 3D meshes, delivering accurate 3D segmentation across diverse meshes and segment descriptions. Specifically, our model leverages the Segment Anything Model (SAM) model to segment the target regions from images rendered from the 3D shape. In light of the importance of the texture for segmentation, we also leverage the pretrained stable diffusion model to generate images with textures from 3D shape, and leverage SAM to segment the target regions from images with textures. Textures supplement the shape for segmentation and facilitate accurate 3D segmentation even in geometrically non-prominent areas, such as segmenting a car door within a car mesh. To achieve the 3D segments, we render 2D images from different views and conduct segmentation for both textured and untextured images. Lastly, we develop a multi-view revoting scheme that integrates 2D segmentation results and confidence scores from various views onto the 3D mesh, ensuring the 3D consistency of segmentation results and eliminating inaccuracies from specific perspectives. Through these innovations, MeshSegmenter offers stable and reliable 3D segmentation results both quantitatively and qualitatively, highlighting its potential as a transformative tool in the field of 3D zero-shot segmentation. The code is available at \url{https://github.com/zimingzhong/MeshSegmenter}.

1 Introduction

MeshSegmenter addresses zero-shot 3D segmentation, where limited annotated 3D data and ambiguity in local mesh regions challenge generalization. It combines 2D segmentation, generated textures, and multi-view revoting to segment text-specified regions across meshes.

1 Introduction

  • Limited accurately annotated 3D data makes training-based methods struggle to generalize to unseen examples, while local mesh regions can be ambiguous across views.The introduction gives black clothing mistaken for hair as an example.Source: §1 · p. 2
  • The framework uses object and region descriptions to guide fine-grained segmentation and demonstrates applications to mesh segmentation and editing.The authors report experimental evaluation across various 3D zero-shot segmentation tasks.Source: §1 · pp. 2–3
  • MeshSegmenter combines 2D zero-shot segmentation of rendered meshes with multi-view aggregation to produce coherent 3D segmentation.The approach aims to use 2D models as a more efficient alternative to handling larger 3D datasets directly.Source: §1 · pp. 2–3
  • Generated textures supplement geometry and help segment regions with weak geometric cues, such as a car door integrated into the body.The framework processes both textured and untextured meshes.Source: §1 · p. 3
  • Face confidence revoting combines detections and segmentations across views to improve 3D consistency and correct perspective-specific errors.The paper describes the resulting segmentation as stable and reliable.Source: §1 · p. 3

2 Related Work

Related work situates MeshSegmenter among zero-shot 2D and 3D segmentation methods, including approaches that adapt 2D models to 3D representations. Prior mesh methods vary in query scope and in how they aggregate predictions across views.

Zero-shot 2D Segmentation

  • Zero-shot 2D segmentation work either aligns image and semantic features for unseen categories or adapts large multimodal semantic spaces to segmentation.The related work describes recent multimodal approaches as improving generalization to unseen categories.Source: §2 · p. 4

Zero-shot 3D Segmentation

  • Zero-shot 3D segmentation includes point-cloud methods and approaches that integrate 2D image-recognition models into 3D shape analysis.NeRF-related work constructs semantic fields and enforces consistency between rendered multi-view labels and ground-truth or estimated labels.Source: §2 · pp. 4–5

Single Query Segmentation

  • 3DHighlighter [9] predicts mesh-vertex probabilities and uses rendered views to align them with a text prompt in CLIP space.Source: §2 · p. 5

Multiple Queries Segmentation

  • SATR [2] aggregates multi-view segmentations by selecting the highest-confidence prompt per facet, while ZSC [1] uses ChatGPT-generated prompts and shape correspondence.The related work says ZSC [1] falters when single-query segmentation is required.Source: §2 · p. 5

3 Method

The method synthesizes textures for meshes, segments rendered views with 2D models, and combines per-view evidence into face-level 3D predictions. Its four stages address texture cues, view coverage, detection and segmentation, and cross-view correction.

3.1 Overview

  • The overview comprises text-guided texture synthesis, 2D zero-shot segmentation, and face confidence revoting for mesh segmentation.It uses both textured and untextured meshes to provide texture and geometric information.Source: §3.1 · p. 6

3.2 Text-Guided Texture Synthesis

  • Stable Diffusion generates mesh textures from the object-class description, adding color cues for regions difficult to localize from shape alone.The text used to prompt texture generation differs from the grounding description used for segmentation.Source: §3.2 · p. 6

3.3 2D zero-shot semantic segmentation

  • Rendered views are processed by GroundingDINO [24] for text-guided boxes and confidence scores, then SAM [18] uses the boxes to produce masks.The camera trajectory samples azimuths at two polar angles to balance segmentation quality with viewpoint coverage.Source: §3.3 · pp. 7–8

3.4 Face Confidence Revoting (FCR)

  • Revoting is designed to prevent erroneous local detections or segmentations from contaminating the final result by using evidence across viewpoints.The method is motivated by errors that can occur in individual views.Source: §3.4 · p. 9
  • Face Confidence Revoting assigns positive confidence to visible masked faces and negative confidence to visible unmasked faces, then sums scores across views.Scores from textured and untextured meshes are averaged, thresholded, and smoothed with neighboring faces.Source: §3.4 · pp. 9–10

4 Experiment

Experiments compare MeshSegmenter with prior zero-shot mesh methods, demonstrate editing and representation extensions, and examine texture synthesis and confidence revoting. The reported evaluation includes a resource-limited subset and a user study.

4.1 Implementation Details

  • The experiments use one Nvidia A40 GPU per object-description and grounding-text case, with Kaolin rendering and GroundingDINO [24] and SAM [18] as the 2D models.The comparison includes 3DHighlighter [9], SATR [2], and official results for non-open-source ZSC [1].Source: §4.1 · p. 10

4.2 Zero-shot Mesh Semantic Segmentation

  • MeshSegmenter outperforms 3DHighlighter [9] and SATR [2] on the ShapeNetPart evaluation and in single- and multiple-query user-study ratings.The evaluation randomly sampled 20 objects per category because multi-view texture generation is computationally resource-intensive; the user study used 20 meshes and 30 respondents.Source: §4.2 · p. 10, Table 1 · p. 11, Table 2 · p. 11

4.3 Application of MeshSegmenter

  • For controlled stylization, MeshSegmenter localizes the prompted hair region for editing, while TEXTure [37] also colors an unspecified ribbon region red.Source: §4.3 · p. 11, Figure 7 · p. 11
  • Text-specified mesh regions can be selected, stretched, warped, or deleted, and mesh segmentation can be transferred to point clouds after optimizing them into mesh structures.The point-cloud extension depends on obtaining a mapping between rendered views and the 3D representation.Source: §4.3 · p. 12, Figure 8 · p. 12, Figure 9 · p. 12

4.4 Ablation Studies

  • Texture synthesis helps reduce the domain gap for 2D models and supports segmentation of regions with weak geometric information.The paper illustrates texture synthesis and face-confidence revoting ablations in Figs. 10 and 11.Source: §4.4 · p. 13, Figure 10 · p. 13, Figure 11 · p. 13
  • The method reports effective segmentation with inaccurate object descriptions, while accurate class descriptions can produce more appropriate generated textures.Source: §4.4 · p. 13

5 Limitation

MeshSegmenter faces limitations in local-level detection by pretrained 2D models and incomplete face visibility across rendered views. It filters whole-object detections and uses fixed-rotation viewpoint sampling to address these challenges.

  • MeshSegmenter segments objects accurately with correct descriptions and also handles homogeneous descriptions such as “animals” and “dogs.”Source: Figure 12 · p. 14
  • Pretrained 2D models are not designed for local-level detection and segmentation, so they can incorrectly detect entire objects as target regions.Source: §5 · p. 14
  • Filtering whole-object detections and face-confidence revoting use correct detections from adjacent viewpoints to correct erroneous detections.Source: §5 · p. 14
  • No rendering strategy guarantees visibility of every face, so the method uniformly samples viewpoints in fixed rotation to maximize the likelihood of seeing more faces.Source: §5 · p. 14

6 Conclusion

MeshSegmenter extends 2D detection and segmentation models to text-specified 3D mesh segmentation. It generates object-description-based textures and combines results across views for stable, consistent segmentation.

  • MeshSegmenter extends 2D detection and segmentation models to segment diverse 3D meshes by text.Source: §6 · p. 14
  • Generating textures based on object descriptions enhances the accuracy of 2D segmentation models.Source: §6 · p. 14
  • A multi-view revoting module integrates views to ensure 3D consistency in segmentation results.Source: §6 · p. 14
  • The framework provides stable and reliable segmentation results and has potential applications in computer graphics and computer vision.Source: §6 · p. 14
Loading 2407.13675v3…