PointCNN运行

1、PointCNN代码地址:https://github.com/yangyanli/PointCNN

作者团队管理的特别好,提出问题会很快回答解决

2、配置环境: Ubuntu16.04+Cuda9+Cudnn7+anaconda(python3.5)+tensorflow1.6(GPU cp35)

参考:https://zhuanlan.zhihu.com/p/34321583

注意:cuda版本9.0     cudnn版本(7.05 for cuda 9.0)

3、以运行ModelNet40 的分类为例

cd ../pointcnn_cls
./train_val_modelnet.sh -g 0 -x modelnet_x3_l4

运行上述代码后,终端显示

Train/Val with setting modelnet_x3_l4 on GPU 0!

表示正在训练,训练的每一步结果储存在     pointcnn文件夹同级       目录下的     /models/cls/ 文件夹中

生成log.txt文件

注:

出现问题,没有生成log.txt,生成了一个“pointcnn_cls_modelnet_x3_l4.txt”

显示AttributeError: 'tuple' object has no attribute 'type'

原因是numpy多个版本错误,参考:https://stackoverflow.com/questions/54200850/attributeerror-tuple-object-has-no-attribute-type-upon-importing-tensorflow

解决方法如下

pip uninstall numpy
sudo apt-get purge python3-numpy

然后要在ubuntu中删除numpy的文件

然后重新安装

pip install numpy

4、用Tensorboard查看训练结果

cd /PointCNN
tensorboard --logdir=../models/cls/

然后终端会返回一个网址,打开即可查看

注:tensorboard  --logdir= 你的log文件所在的目录,不是log文件是目录

你可能感兴趣的:(PointCNN运行)