admin 管理员组

文章数量: 1184232

Win10系统下训练RandLA-Net点云语义分割模型

  • 1、编译cpp模块
    • 1.1、编译下采样模块
    • 1.2、编译最近邻搜索模块
  • 2、数据预处理
  • 3、训练和测试

环境
tensorflow:2.2.0
python:3.7
cuda:10.1
cudnn:7.6
GPU:2080TI

1、编译cpp模块

1.1、编译下采样模块

cmd中进入cpp_wrappers\cpp_subsampling
执行命令

python setup.py build_ext --inplace

编译后会生成grid_subsampling.cp37-win_amd64.pyd文件。

1.2、编译最近邻搜索模块

cmd中进入nearest_neighbors
执行命令

python setup.py build_ext --inplace

编译后会生成nearest_neighbors.cp37-win_amd64.pyd文件。
虽然我们能够编译成功,但是在实际使用的时候该模块搜索出来的最邻近点是错误的,这也导致了之前遇到的在windows系统下训练randlanet模型精度特别低的问题。<

本文标签: 语义 模型 系统 NET RandLA