BAAF-Net训练S3DIS数据集
- 一、数据集预处理
- 二、BAAF-Net
- 2.1、编译下采样和最近邻搜索模块
- 2.2、训练
- 2.3、交叉验证
系统:Ubuntu
tensorflow:2.2.0
cuda:10.2
一、数据集预处理
S3DIS数据集下载和清洗参考文章在tensorflow2.0环境下使用RandLA-Net训练S3DIS数据集。
二、BAAF-Net
2.1、编译下采样和最近邻搜索模块
sh compile_op.sh
编译时出现的问题如下:
应该是我系统中安装的是cuda10.2,所以无法找到cuda10.0/bin/nvcc这个文件。可以将cuda10.2这个文件夹软链接到cuda10.0.
ln -s cuda-10.2 cuda-10.0