Pytorch實現PointNet中的點雲分類網絡。

下面是PointNet論文中分類模型的結構： 但是對於模型的細節，PointNet論文中並沒有詳細的解釋，尤其是T-Net，可以參考PointNet的supplemental部分。如果找不到，可以留言找我要。 話不多說，下面是代碼，基本上完全還原了論文中的PointNet分類模型。 第一部分：數據處理模塊 更新一下代碼，修復了原先batchsize=1的時候會出錯的毛病。 import h5py

>>阅读原文<<