腾讯混元3D团队开源 P3-SAM: 原生3D零件分割

将3D资产分割为其组成部分对于增强三维理解、促进模型复用以及支持部件生成等应用至关重要。然而,现有方法在处理复杂物体时存在鲁棒性不足等问题,且无法实现全自动化流程。本文提出了一种原生3D点提示部件分割模型P3-SAM,旨在实现任意三维物体的全自动组件分割。受SAM启发,P3-SAM由特征提取器、多重分割头和IoU预测器组成,支持用户交互式分割。我们还提出了一种算法来自动筛选并合并模型预测的掩膜,实现部件实例分割。该模型在我们新建的包含370万个带合理分割标签模型的数据集上进行训练。对比实验表明,本方法能在任意复杂物体上实现精确分割结果和强大鲁棒性,达到了当前最优性能。

方法


P3-SAM的网络架构:输入点云被送入特征提取器以获取逐点特征。这些特征、点提示以及原始点云随后被输入到一个两阶段多掩模分割器中,生成不同尺度的三个掩模。最后,利用IoU预测器评估掩模质量,并选择最佳掩模作为最终预测结果。


自动分割流程:通过FPS采样点提示,输入P3-SAM模型获取多个掩码,随后采用非极大值抑制(NMS)合并冗余掩码。最后将点级掩码投影到网格面上,得到部件分割结果。

https://murcherful.github.io/P3-SAM/

相关推荐
shayudiandian9 分钟前
用PyTorch训练一个猫狗分类器
人工智能·pytorch·深度学习
这儿有一堆花13 分钟前
把 AI 装进终端:Gemini CLI 上手体验与核心功能解析
人工智能·ai·ai编程
子午26 分钟前
【蘑菇识别系统】Python+TensorFlow+Vue3+Django+人工智能+深度学习+卷积网络+resnet50算法
人工智能·python·深度学习
模型启动机36 分钟前
Langchain正式宣布,Deep Agents全面支持Skills,通用AI代理的新范式?
人工智能·ai·langchain·大模型·agentic ai
Python私教41 分钟前
别让 API Key 裸奔:基于 TRAE SOLO 的大模型安全配置最佳实践
人工智能
Python私教43 分钟前
Vibe Coding 体验报告:我让 TRAE SOLO 替我重构了 2000 行屎山代码,结果...
人工智能
prog_610344 分钟前
【笔记】和各大AI语言模型写项目——手搓SDN后得到的经验
人工智能·笔记·语言模型
zhangfeng11331 小时前
深入剖析Kimi K2 Thinking与其他大规模语言模型(Large Language Models, LLMs)之间的差异
人工智能·语言模型·自然语言处理
paopao_wu1 小时前
人脸检测与识别-InsightFace:特征向量提取与识别
人工智能·目标检测
Aevget1 小时前
MyEclipse全新发布v2025.2——AI + Java 24 +更快的调试
java·ide·人工智能·eclipse·myeclipse