RandLA-Net 调通代码1：Windows 11 + Tensorflow 2.4 + S3DIS数据集【踩坑+填坑】

目  录

一、克隆源代码

二、安装 Tensorflow

三、一些重要的 lib

四、邻近搜索和下采样 sh compile_op.sh【具体命令如下】

4.1. KNN

4.2. subsampling

五、准备S3DIS数据

5.1. 下载 Stanford3dDataset_v1.2_Aligned_Version.zip

5.2. 放到 /data/S3DIS[手动新建文件夹]，并解压到当前文件夹

5.3. 运行 python utils/data_prepare_s3dis.py 过程中的坑

5.4. 结果

六、交叉【训练】【验证】：sh jobs_6_fold_cv_s3dis.sh

6.1 【训练】区域5：运行 python -B main_S3DIS.py --gpu 0 --mode train --test_area 5 过程中的坑

6.2 【测试】区域5：运行 python -B main_S3DIS.py --gpu 0 --mode test --test_area 5 过程中的坑

七、计算mIoU与可视化

7.1. 移动文件【为了和源代码保持一致】

7.2. 运行 python utils/6_fold_cv.py 过程中的坑

7.3. 如果 visualization = True【不建议】

7.4. visualization = True，利用laspy生成预测的las文件 Classification=labels, UserData=pred

7.4.1. 添加laspy生成las文件的函数

7.4.2. 修改[utils/6_fold_cv.py]

---

一、克隆源代码

克隆大佬的源代码 https://github.com/QingyongHu/RandLA-Net

cd RandLA-Net

二、安装 Tensorflow

找配套的python、CUDA和cudnn。TF现在是真lj。

本文环境：python = 3.8 + CUDA = 11.0 + cudnn = 8.0

【pip3 install tensorflow-gpu==2.4.4】

三、一些重要的 lib

【pip install cython==0.29.15】

【【pip install -U scikit-learn】】

【pip install pandas==0.25.3】

四、邻近搜索和下采样 sh compile_op.sh【具体命令如下】

【pip install cython==0.29.15】

Microsoft Visual C++ 14.0 or greater is required

4.1. KNN

cd utils/nearest_neighbors

python setup.py install --home="."

cd ../../

结果：[/utils/nearest_neighbors] 下生成 四个文件夹和knn.cpp

4.2. subsampling

cd utils/cpp_wrappers/cpp_subsampling

python setup.py build_ext --inplace

cd ../../../

结果：[/utils/cpp_wrappers/cpp_subsampling] 下生成 build文件夹和grid_subsampling.cp38-win_amd64.pyd

五、准备S3DIS数据

5.1. 下载 Stanford3dDataset_v1.2_Aligned_Version.zip

5.2. 放到 /data/S3DIS[手动新建文件夹]，并解压到当前文件夹

【【pip install -U scikit-learn】】

【pip install pandas==0.25.3】

5.3. 运行 python utils/data_prepare_s3dis.py 过程中的坑

坑1：[helper_tool.py] 中的 from open3d import linux as open3d 报错 ModuleNotFoundError: No module named 'open3d'

填坑1：【open3d仅用于最后的可视化，不安装也可以，把相关的代码隐掉就行。】【十分推荐】
        将[helper_tool.py] 导入open3d的代码隐掉；将 draw_pc 函数的内部代码隐掉

填坑2：【以下是不隐代码的解决方法】【十分不推荐】
        【pip install open3d==0.18】
        ......解决lib的版本问题:numpy==1.19.5；typing-extensions~=3.7.4；dash==2.6.0......
        将[helper_tool.py]中的 from open3d import linux as open3d 改为 import open3d

坑2：No module named 'nearest_neighbors.lib.python.nearest_neighbors'

填坑：/utils/nearest_neighbors/lib/python 下的文件夹：KNN_NanoFLANN-0.0.0-py3.8-win-amd64.egg 名称改为 KNN_NanoFLANN
填坑：将[helper_tool.py] 中的 import nearest_neighbors.lib.python.nearest_neighbors as nearest_neighbors
    改为 import utils.nearest_neighbors.lib.python.KNN_NanoFLANN.nearest_neighbors as nearest_neighbors

填坑：将[helper_tool.py] 中的 import cpp_wrappers.cpp_subsampling.grid_subsampling as cpp_subsampling
    改为 import utils.cpp_wrappers.cpp_subsampling.grid_subsampling as cpp_subsampling

坑3：[WinError 3] 系统找不到指定的路径。: '/data/S3DIS\\original_ply'

填坑：[utils/data_prepare_s3dis.py]中的 dataset_path 改成绝对路径。
    dataset_path = 'C:/xinjiang/RandLA-