qecnetworks:ECCV 2020口头论文“用于3D点云的四元数等效胶囊网络”的正式实施 源码

用于3D点云的四元数等变胶囊网络 通过创建, , ,, , 。 该存储库包含 (QEC-Net)的实现。 特别是,我们发布了用于训练和测试QEC-Net的代码,以对3D形状进行分类和相对旋转估计,并发布了预训练的模型以快速复制结果。 有关QEC-Net的直观说明,请查看。 有关源代码,请访问。 抽象 我们提出了一种用于处理点云的3D胶囊模块,该模块与3D旋转和平移以及输入点的排列相同。 运算符接收从输入点云计算而来的一组稀疏的局部参考系,并通过一种新颖的四元数动态路由过程来建立端到端的变换等方差。 此外,我们在理论上将胶囊之间的动态路由连接到著名的Weiszfeld算法,这是一种解决具有