UniCo
UniCo: Unified Primitive Proxies for Structured Shape Completion(CVPR 2026, arXiv:2601.00759)— 基元代理查询在共享特征上并行解码点补全和结构化基元。
Overview
- 发布时间:CVPR 2026(arXiv 2601.00759)
- 关键词:基元代理 · 结构化补全 · 可学习查询 · 匈牙利匹配
- 核心问题:点云补全不仅是恢复形状,还需结构化理解——识别物体由哪些几何基元(平面、圆柱、球体等)组成。现有方法忽略基元层面的结构信息。
架构
Unified Primitive Proxies(统一基元代理)
- K=40 个可学习查询,在共享特征空间上并行解码
三个预测头
- 语义头(Semantic Head):预测每个基元的语义类别
- 几何头(Geometric Head):预测基元的几何参数(位置、朝向、尺度)
- 成员关系头(Membership Head):预测每个点属于哪个基元
在线目标更新 + 匈牙利二部匹配
- 训练过程中动态更新基元代理的目标分配
- 匈牙利算法做最优匹配,避免基元之间的竞争冲突
并行解码
- 点补全分支:恢复稠密点云几何
- 基元分解分支:输出 K 个结构化基元表示
- 两个分支共享同一套代理特征,天然互增强
性能
| 指标 | CD | NC |
|---|---|---|
| UniCo | 降低 50% | 提升 7% |
| 对比基线 | — | — |
- CD(Chamfer Distance)降低 50%
- NC(Normal Consistency)提升 7%
- 同时输出点云和结构化基元表示
关键创新
- 基元代理查询:首次将可学习基元查询用于点云补全
- 三头并行预测:语义 + 几何 + 成员关系统一建模
- 在线目标更新:动态匈牙利匹配解决基元分配歧义
Related
- point-cloud-completion — 点云补全总览
- hyper-pcn — Hyper-PCN 超图点云补全(高阶相关性建模)
- x-part — X-Part 可控部件级 3D 生成
Sources
- arXiv:2601.00759 — UniCo, CVPR 2026