资源相关
论文源码 地址:https://github.com/charlesq34/pointnet
数据集下载地址:https://shapenet.cs.stanford.edu/media/modelnet40_ply_hdf5_2048.zip
源码使用TensorFlow框架,并在Python 2.7, TensorFlow 1.0.1, CUDA 8.0 and cuDNN 5.1 on Ubuntu 14.04 进行测试
我们使用环境为 Python3.6, TensorFlow 1.11.0 CUDA 9.0 and cuDNN 7.3.1 on Ubuntu 16.04
环境配置可以参考:
[机器学习] 1:最简单最直接Ubuntu16.04+CUDA9.0+cuDNN7.3+1080Ti驱动配置
[机器学习] 2: Linux下Anaconda+TensorFlow-gpu+Pycharm版本配置
代码运行
- python train.py 进行训练 (提前将 数据集.zip 放到 train.py 同一目录下 或者 在这里运行时,自动从网络下载)
- tensorboard --logdir log (使用tensorboard 来查看网络结构 与 训练过程)
- python evaluate.py --visu (训练结束后 ,进行评估与可视化)
ps: No module named ‘PIL’ 提示缺少模块 使用pip install pillow安装即可