目 录
四、邻近搜索和下采样 sh compile_op.sh【具体命令如下】
5.1. 下载 Stanford3dDataset_v1.2_Aligned_Version.zip
5.2. 放到 /data/S3DIS[手动新建文件夹],并解压到当前文件夹
5.3. 运行 python utils/data_prepare_s3dis.py 过程中的坑
六、交叉【训练】【验证】: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 过程中的坑
7.2. 运行 python utils/6_fold_cv.py 过程中的坑
7.3. 如果 visualization = True【不建议】
7.4. visualization = True,利用laspy生成预测的las文件 Classification=labels, UserData=pred
一、克隆源代码
克隆大佬的源代码 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-