一、降级Tensorflow
Kaggle安装的tensorflow版本为2.1.0,而PointNet的代码是针对tensorflow1.0编写的,所以直接运行会存在兼容性问题,例如出现报错AttributeError: module 'tensorflow' has no attribute 'placeholder'
。解决方法有两种,一是将代码中的import tensorflow as tf
改成import tensorflow.compat.v1 as tf
tf.disable_v2_behavior()
,但这种方法存在一定局限性,以PointNet为例,在运行train.py时会调用运行其他.py文件,所以仅仅修改train.py文件中的代码并不能解决问题,修改所有.py文件工作量又太大。所以推荐的方法是对当前运行环境的tensorflow版本进行降级(输入!pip list
可以查看当前环境中各个库的版本信息)
!pip install tensorflow==1.5.0
二、从GitHub导入文件
!git clone https://github.com