复现基于Tensorflow的Randla-Net（附踩坑过程）

原创

已于 2023-05-29 13:27:56 修改 · 2.8k 阅读

标签

#tensorflow #python #ubuntu

收录于

于 2023-05-29 13:27:00 首次发布

文章讲述了在配置RandLA-Net增强版博客所需的环境过程中遇到的问题，包括数据集下载、CUDA和Tensorflow版本匹配以及open3d版本兼容性。最终采用python3.6+Tensorflow2.6+CUDA11.4的组合，并对代码做了一些修改以适应Tensorflow2.0的行为，成功运行了项目。

博客基于的github资源：RandLA-Net-Enhanced

1.数据下载

代码所基于的是 SemanticKITTI公开数据集，官方链接为：SemanticKITTI - A Dataset for LiDAR-based Semantic Scene Understanding

下载图中所示几个文件，解压后将其中的dataset文件夹拷贝至/data/semantic_kitti/dataset文件夹，选择合并文件夹即可。

由于数据集网站是国外地址，可以找合适的百度网盘资源进行下载。

2.环境配置

2.1 第一次尝试

原作者给出的环境配置方案如下：

conda create -n randlanet python=3.5
source activate randlanet
conda install cudatoolkit=9.0 cudnn=7.3.1
pip install -r helper_requirements.txt
sh compile_op.sh
pip install tensorflow-gpu==1.11

由于实验室服务器系统为Ubuntu