当前位置:   article > 正文

pointnet++复现

pointnet++复现

首先修改代码中所有错误,与自己的python和tensorflow适配,导入该导入的包,其中provider和eulerangles和plyfile都在utils文件下或者pointnet-master的代码中
修改tf_ops中的文件:参考第一篇第二篇
注意:报错

/usr/bin/ld: cannot find -ltensorflow_framework,collect2: error: ld returned 1 exit status。
  • 1

应该是没有修改sh文件时未能找到/usr/…/ltensorflow_framework这个路径
解决方法:
我是用的云服务器,将sh文件中的-ltensorflow_framework修改为-l:libtensorflow_framework.so.1
有的人可能是要改成-libtensorflow_framework
然后运行

cd tf_ops/sampling
sh tf_sampling_compile.sh
cd tf_ops/grouping
sh tf_grouping_compile.sh
cd tf_ops/d_interpolation
sh tf_interpolate_compile.sh
  • 1
  • 2
  • 3
  • 4
  • 5
  • 6

进入环境

conda activate tf
  • 1

训练

cd autodl-nas/pointnet2-master
  • 1
CUDA_VISIBLE_DEVICES=0   python  train.py
  • 1

报错

tensorflow.python.framework.errors_impl.NotFoundError: /root/autodl-nas/pointnet2-master/tf_ops/sampling/tf_sampling_so.so: undefined symbol: _ZN10tensorflow12OpDefBuilder4AttrESs
  • 1

将sh文件中的-D_GLIBCXX_USE_CXX11_ABI=0改为1
再运行.sh文件

cd tf_ops/sampling
sh tf_sampling_compile.sh
cd tf_ops/grouping
sh tf_grouping_compile.sh
cd tf_ops/d_interpolation
sh tf_interpolate_compile.sh
  • 1
  • 2
  • 3
  • 4
  • 5
  • 6

在这里插入图片描述

训练结果可视化

tensorboard --logdir=log#这里要指定日志文件夹名字,因为默认似乎是logs,但是pointnet++分类训练生成的日志文件名为log
  • 1

在这里插入图片描述系统架构图
在这里插入图片描述
在这里插入图片描述
在这里插入图片描述

训练后验证结果

CUDA_VISIBLE_DEVICES=0 python evaluate.py --num_votes 12
  • 1

在这里插入图片描述

可视化(此步骤无需gpu)

在utils文件下

sh compile_render_balls_so.sh
python show3d_balls.py
  • 1
  • 2

获得可视化结果
在这里插入图片描述

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/你好赵伟/article/detail/101761?site
推荐阅读
相关标签
  

闽ICP备14008679号