Point-Transformers:点变压器

各种点变压器的Pytorch实现 近来，各种方法将变压器应用于点云： ， ， 。 此存储库是这些方法的pytorch实现，旨在在公平的环境下对它们进行比较。 当前，这三种方法都已实现，同时调整了它们的超参数。 分类 资料准备 下载路线ModelNet并保存在modelnet40_normal_resampled 。 跑步 更改要在config/config.yaml使用的方法并运行 python train.py 结果 使用Adam的学习速率每50个周期衰减0.3，训练200个周期； 数据扩充遵循。 对于横双和Nico，初始LR为1e-3（如果有人可以微调这些超参数，我将不胜感激）； 对于孟浩，建议的初始LR为1e-4。 下面列出了ModelNet40分类结果（实例平均值）： 模型 准确性 横双 89.6 孟浩 92.6 尼科 85.5 各种各样的 一些代码和培训设置是从借用的。