linux 下跑通pointnet++网络模型

linux 下跑通pointnet++网络模型

pointnet++模型下载地址：https://github.com/charlesq34/pointnet2

编译环境
采用anaconda3建立虚拟环境，在环境下进行配置运行：
创建自己的环境 conda create -n name python=2.7 用的python2.7的环境
source activate name;进行激活
接着进行一下模块安装
tensorflow-gpu1.14.0+cudnn 7.6.5+cuda10.1+cython+h5py
安装好以上后需对pointnet2-master中tf_ops文件中的三个文件进行修改

分别对三个文件中的.sh文件进行修改后编译：
如对sampling文件中的sampling_compile.sh进行修改

1，3，5环境查找可用一下命令
python -c ‘import tensorflow as tf; print(tf.sysconfig.get_include())’ （1）
python -c ‘import tensorflow as tf; print(tf.sysconfig.get_lib())’ （3)
(5)根据（1）补充自己路径
GCC高于4.8版本时在-D_GLIBCXX_USE_CXX11_ABI=0 前面加#注释

修改后进行运行 sh sampling_compile.sh进行编译

编译时可能会出现找不到-ltensorflow_framework的情况

解决方法：libtensorflow_framework.so与libtensorflow_framework.so.1进行映射

cd 进入libtensorflow_framework.so目录下：库目录可在（5）下查找；
运行 ln -s libtensorflow_framework.so.1 libtensorflow_framework.so

f_ops文件中的另外两个文件也是一样的修改方式
编译成成后会生成tf_sampling_so.so，tf_sampling_g.cu.o文件

以上完成后进入到pointnet-master目录下运行
python train.py
查看其它运行参数设置可运行python train.py -h
如果显存不够可以在train.py修改max_epoch

运行成功后出现以下