PoinTr

收录理由

三维扫描仪往往只能捕捉到物体的一部分，PoinTr 就是用来补全那些缺失区域的。它将无序点云看作带位置编码的点代理序列，再交给 Transformer 编码器-解码器处理，对于想把 Transformer 用到文本和图像以外场景的研究者来说，是个很有参考价值的例子。仓库里除了核心权重，还提供了多个基准上的预训练模型、改进版 AdaPoinTr 以及 SnowFlakeNet 的实现，既能直接作为基线，也能看出这套方法的演进脉络。安装方面，给出了 CUDA 扩展的编译脚本，以及作者提出的 ShapeNet-55/34 数据集的使用说明，研究人员照着就能复现论文里的数字，或者把结构改到自己的流程里，不用从头搭工具。