PointNet复现
PointNet论文地址
参考:1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19
1. 预备工作
# 下面两行用来安装h5py
sudo apt-get install libhdf5-dev
sudo pip install h5py
source ~/tf101py27/bin/activate #激活进入虚拟环境
# 或者使用如下conda方式下载
conda activate py35
conda install libhdf5-dev
conda install h5py
# 下面四行用来找到Tensorflow包括路径和库路径。
python
import tensorflow as tf
print(tf.sysconfig.get_include())
print(tf.sysconfig.get_lib())
2. 分类
分类问题就是判断三维模型属于哪一类,如汽车、桌子、书等
2.1 训练
- 下载modelnet40_ply_hdf5_2048放置到pointnet-master/data文件夹下(provider.py 中设置了DATA_DIR 为数据存放的位置)
- 执行train.py
cd /d D:\pointnet-master
python train.py
- 部分结果