RandLA-Net 复现

GPU3090 + CUDA12

1、代码 [github地址]

git clone --depth=1 https://github.com/QingyongHu/RandLA-Net && cd RandLA-Net

2、虚拟环境中配置：

在跑代码的时候出现错误：open3d.so文件中函数报错。查看open3d版本发现不是要求的0.3版本，在安装旧版本时发现python3.7不再支持旧版本。因此考虑降版本。[参考]

刚开始装python3.7,报错，改成python3.6成功！

conda create -n randlanet python=3.6
source activate randlanet

cuda11.3

conda search cudatoolkit --info
conda install cudatoolkit=11.3

cudnn:

conda search cudnn --info
conda install cudnn=8.2.1

选择tensorflow-2.6.0[参考][tersorflow版本对应]

pip install tensorflow-gpu==2.6 -i https://pypi.tuna.tsinghua.edu.cn/simple  --timeout=120

 最后环境配置完成：

pip install -r helper_requirements.txt  
sh compile_op.sh

准备数据集，将data_prepare_s3dis.py路径修改为自己数据集路径，执行下面代码生成input_0.040和original_ply

 python utils/data_prepare_s3dis.py

 

 

 

 

 

由于tensorflow是2.0版本，因此代码中部分位置需要修改：

将main_Semantic3D.py、RandLANet.py、helper_tf_util.py文件中的

import tensorflow as tf

修改为：

import tensorflow.compat.v1 as tf
tf.disable_v2_behavior()