Changfeng Ma Yang Li Xinhao Yan Jiachen Xu Yunhan Yang Chunshi Wang Zibo Zhao Yanwen Guo Zhuo Chen Chunchao Guo

摘要

将3D资产分割为其组成部分对于提升3D理解能力、促进模型复用以及支持各类应用（如部件生成）至关重要。然而，现有方法在处理复杂物体时存在鲁棒性差的问题，且难以实现全流程自动化。本文提出一种原生支持3D点提示的部件分割模型——P3-SAM，旨在实现任意3D物体的全自动部件分割。受SAM（Segment Anything Model）启发，P3-SAM由特征提取器、多个分割头以及一个IoU预测器组成，支持用户交互式分割。此外，我们设计了一种算法，可自动选择并合并模型预测的掩码，以实现部件实例分割。该模型在新构建的大型数据集上进行训练，该数据集包含近370万组带有合理分割标签的3D模型。实验结果表明，我们的方法在各类复杂物体上均能实现高精度分割，并展现出强大的鲁棒性，达到当前最优性能。相关代码即将开源。