当前位置：首页 > news >正文

三维点云深度学习代码torch-points3d-SiamKPConvVariants复现记录（持续更新中）

news 2025/5/31 11:48:20

参考文献：de Gélis I, Lefèvre S, Corpetti T. Siamese KPConv: 3D multiple change detection from raw point clouds using deep learning[J]. ISPRS Journal of Photogrammetry and Remote Sensing, 2023, 197: 274-291.

Siamese KPConv: 3D multiple change detection from raw point clouds using deep learning - ScienceDirecthttps://www.sciencedirect.com/science/article/pii/S0924271623000394

插句题外话：IdeGelis这家伙是真牛B（‪Iris de Gélis‬ - ‪Google 学术搜索‬），深耕遥感和深度学习领域，感觉很年轻的亚子， ISPRS和TGRS真是咔咔发（顶级膜拜大佬😜😜😜）还发布了广为流传的Urb3DCD数据集（需要数据集的铁子，点赞来波关注，网盘提取码双手送上😁😁😁）苦苦等待许久下载到的，求个关注不过分吧😋

吐槽开始：大佬牛是牛，但咱就是说能不能把GitHub上的操作步骤写得再详细一点😭，按照简单的教程运行一步错一步，报错都看不明白。网上能参考的资料真是没有，于是我鼓起勇气发邮件直接联系作者（静静等待回复中。。。）：

复现的心路历程：

conda create --prefix /CD/envs/SKP38 python=3.8
conda activate /CD/envs/SKP38
降级：pip install pip==24.0 -i https://pypi.tuna.tsinghua.edu.cn/simple（到底需不需要降级？看自己）
pip安装requirements.txt中的工具包
cd /CD/torch-points3d-SiameseKPConv-master
pip install -r requirements.txt -i https://pypi.tuna.tsinghua.edu.cn/simple（有时工具版本不匹配，就会报错，报错我是习惯了，直接无视，以后缺什么再按什么。强迫症除外😱）

nvcc --version 查看电脑的CUDA版本
pip install torch==2.1.0+cu121 torchvision==0.16.0+cu121 torchaudio==2.1.0 -f https://download.pytorch.org/whl/torch_stable.html

pip install torch-scatter -f https://data.pyg.org/whl/torch-2.1.0+cu121.html
pip install torch-sparse -f https://data.pyg.org/whl/torch-2.1.0+cu121.html
pip install torch-cluster -f https://data.pyg.org/whl/torch-2.1.0+cu121.html
pip install torch-spline-conv -f https://data.pyg.org/whl/torch-2.1.0+cu121.html
pip install torch-geometric
pip install zipp==3.15.0
pip install hydra-core==1.0.1 omegaconf==2.2.2
pip install wandb==0.15.12
pip install numba==0.53.1 scikit-learn==0.24.2

刚开始选择shapenet数据集、模型pointnet2、分割Chair、不开启wandb （从我认为最简单的开始搞搞搞）

HYDRA_FULL_ERROR=1 python train.py task=segmentation models=segmentation/pointnet2 model_name=pointnet2 data=segmentation/shapenet +data.category=Chair training.wandb.public=false

出现如下画面，恭喜你训练开始了，也算是看到太阳了😎😎😎