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

关注

VIP免费原创 2024-10-03 19:58:53 2.4k 阅读 AI 写同款· GEO 优化 ›

【计算机视觉】简述对RandLA-Net（大场景点云）的理解之前讲述的都是基于逐点MLP和点卷积的文章，对于逐点MLP的文章来说（PointNet和PointNet++）都是采用了对称函数（maxpooling）的做法，这样做的目的是为了解决点云的无序性，但是最大的不足就是他们只提取了全局特征（global feature），丢失了很多信息，虽然PointNet++设计了set abstraction层，用于提取周围点特征，但是提取的效果不好，而且计算量比较差，对于点卷积的文章，他们利用Kernel对领域点进行聚合，然后提取特征并且进行分割和分类，这样做的好处就是解阅读详情

目录

一、克隆源代码

二、安装 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

Win和Linux不一样

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-Net/data/S3DIS/Stanford3dDa