本文主要是介绍Stratified Transformer复现和调试运行记录,训练S3DIS数据集(点云语义分割),希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!
环境:
ubuntu20.04,gcc和g++都是7.5,cuda是11.3,显卡3090
创建虚拟环境:
python=3.7,离线安装torch=1.8.0,torchision=0.9.0 (torch 不要超过1.12.1我的编译时报错) 安装torch=1.8.0(兼容性稍微好点)是因为torch.points3d这些库的要求,也可以使用作者用的torch1.7.1,cuda10.1,我的torch.points3d编译又报错了,哭
torch和torchision按好后直接怼图中红色的框,按顺序,编译torch_sparse和point3d估计用了快一个小时,在安装没有安装的库h5py==3.2.1,SharedArray==3.2.1,最后,注意不要随意安装numpy==1.19.5,可能会报错numpy.core什么的,这个还挺难的,万一出现了用conda和pip卸载conda uninstall numpy 和pip install numpy,中间使用conda clean -p 清理包,再用conda install numpy,这是一个可借鉴方法,不是百试百灵。
最后,祝你好运!!!
11.20 补充一下在老师那儿跑的结果,S3DIS,在区域5上,3090训练完成时间40小时左右,best Iou=67.2,应该是平均Iou ,接下来,就是按部就班地跑6折交叉验证了!
这篇关于Stratified Transformer复现和调试运行记录,训练S3DIS数据集(点云语义分割)的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!