最近做的工作是点云识别,在网上找了几个库,PointCNN,PointNet,PointNet++,PointCNN和PointNet++最终因为过于不好运行而被我放弃.PointNet说是识别不够精确,但是对于我来说,一般的也就够用了.不过由于我们现在的项目是由C++编写的,而PointNet是由多个python库构成,后续肯定会存在一些问题,先不管这么多了,能跑起来再说.
环境配置
1.tensorflow的安装
操作系统是Ubuntu18.04,python版本3.6.9.(我电脑上存在两个版本的Python,2.7和3.6.9,python默认调用2.7版本,想要调用3.6.9版本,需要运行Python3)
1.如果安装了pip,进行下一步.
sudo apt update
sudo apt install python3-dev python3-pip
sudo pip3 install -U virtualenv # system-wide install
2.请从 PyPI 中选择以下某个 TensorFlow 软件包进行安装:
tensorflow
:支持 CPU 和 GPU 的最新稳定版(适用于 Ubuntu 和 Windows)。tf-nightly
:预览 build(不稳定)。Ubuntu 和 Windows 均包含 GPU 支持。tensorflow==1.15
:TensorFlow 1.x 的最终版本。
我安装的是1.15 ,之前安装2.0版本的很多代码是不兼容的.
pip3 install --user --upgrade tensorflow_gpu==1.15
3.验证安装
python3 -c "import tensorflow as tf; print(tf.reduce_sum(tf.random.normal([1000, 1000])))"
2.cuda安装
1.首先查看你的显卡是否支持cuda安装,点此查询列表中查询是否包含你所用的显卡型号.如果支持,进行下一步.
2.根据你的系统选择索要下载的版本.下载链接
3.下载PointNet库
1.下载完毕后,运行如下代码
python3 train.py
运行效果如下:(有人说运行了十个小时,我等运行完再来更新)....
参考:https://www.tensorflow.org/install/pip?hl=zh-cn&lang=python3#%E7%B3%BB%E7%BB%9F%E5%AE%89%E8%A3%85