sh4175515的博客

原创 基于MindSpore的PointTransformer代码实现

作者借鉴了在自然语言处理领域取得巨大成就的Transformer模块，将其应用于三维点云处理领域的目标检测和语义分割。Point Transformer 模型框架如下图所示。本文主要设计了Transformer、TransitionDown和TransitionUp模块，并结合MLP在ModelNet40和ShapeNet数据集上取得了很好的效果。Transformer模块促进局部特征向量之间的信息交换，为所有数据点生成新的特征向量作为输出，并考虑特征向量的内容及其在三维空间中的分布。