python3.6+tensorflow1.14
AutoDL云服务器:RTX2080ti
准备
conda activate tf
安装包open3d-python
pip install open3d-python==0.3.0`
缺少plyfile,我是复制粘贴pointnet中utils下的plyfile.py到3D-BoNet-master中
cd autodl-nas/3D-BoNet-master
编译tf_ops
修改tf_ops下三个文件夹中的.sh文件路径修改参考以下两个链接中tf_ops部分:
参考1和参考2
修改.sh文件后编译文件
cd tf_ops/sampling
chmod +x tf_sampling_compile.sh
./tf_sampling_compile.sh
cd tf_ops/grouping
chmod +x tf_grouping_compile.sh
./tf_grouping_compile.sh
cd tf_ops/interpolation
chmod +x tf_interpolate_compile.sh
./tf_interpolate_compile.sh
训练
CUDA_VISIBLE_DEVICES=0 python main_train.py> train.log 2>&1; shutdown
测试
CUDA_VISIBLE_DEVICES=0 python main_eval.py> train.log 2>&1; shutdown
可视化
待更新ing