Pytorch实现PointNet中的点云分类网络。

下面是PointNet论文中分类模型的结构： 但是对于模型的细节，PointNet论文中并没有详细的解释，尤其是T-Net，可以参考PointNet的supplemental部分。如果找不到，可以留言找我要。 话不多说，下面是代码，基本上完全还原了论文中的PointNet分类模型。 第一部分：数据处理模块 更新一下代码，修复了原先batchsize=1的时候会出错的毛病。 import h5py

>>阅读原文<<