3D-BoNet复现

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