一、pytorch版本KPConv模型github地址
二、系统环境
系统:win10
torch:1.7
cuda:10.1
cudnn:7.6.5
显卡:2080TI
三、bug修改
3.1、编译下采样和最近邻搜索模块bug
作者给的编译方法是
在win10系统下,分别编译cpp_neighbors和cpp_subsampling,执行
cpp_wrappers/cpp_neighbors/build.bat
和
cpp_wrappers/cpp_subsampling/build.bat
我自己在win10系统下测试时发现执行build.bat会提示没有py命令,如果你也遇到同样的问题,可以按照下面的操作编译
编译cpp_neighbors,在cmd下进入cpp_warppers\cpp_neighbors文件夹,执行下面语句
python setup.py build_ext --inplace
编译cpp_subsampling的方法和上面一样。
3.2、读取数据使用多线程bug
修改config文件不使用多线程
在train_S3DIS.py的第62行
input_threads=