1.前提准备
首先说一下我的电脑配置
软件:
linux16.04
anaconda+python3.6+tensorflow1.14
硬件:
cpu 8GB
根据官方文档:pointnet.github需要先安装h5py:
sudo apt-get install libhdf5-dev
sudo pip install h5py
然后下载pointnet源码,下面具体讲复现步骤。
2.pointnet分类网络
2.1 训练
分类用的数据集为modelnet40,该数据集含有40个CAD模型,运行train.py可以自动下载到data文件夹中。如果提示下载失败,可以自己先下载好数据集。
代码默认batchsize为32,我是用CPU训练,内存有限所以把该参数调小了。据说用pointnet_cls_basic可以减少训练时间(还没有验证)。