PointNet复现
设备及运行环境基本情况:
(1)硬件:ThinkPad P53移动图像工作站,CORE i7 -9850H 32G内存,显卡为NVIDIA Quadro T2000 4G ;
(2)软件:Ubuntu18.04、Python2.7、Tensorflow1.13.1, CUDA 10.0 and cuDNN 7.5.0。
1.准备工作
#To install h5py for Python
sudo apt-get install libhdf5-dev
sudo pip install h5py
# or using the way of conda to install
conda activate py35
conda install libhdf5-dev
conda install h5py
2. 训练与测试
PointNet主要应用在三种类型的任务中,包括对象分类(object classification)、部件分割(part segmentation)和场景分割(scene segmentation)。
2.1分类
-
数据准备:
分类训练数据为ModelNet40,包括40种人造的对象类型,共有12311个CAD模型,训练时需将其转为h5文件,可以直接下载好h5文件modelnet40_ply_hdf5_2048.z