文件名称:emd的matlab代码详解-caspr:NeurIPS2020论文“CaSPR:学习规范时空点云表示”的代码
文件大小:4.64MB
文件格式:ZIP
更新时间:2024-06-22 03:12:36
系统开源
emd的matlab代码详解CaSPR:学习规范时空点云表示 这是 NeurIPS 2020 聚光灯论文的官方实现。 有关更多信息,请参阅 . 环境设置 注意:此 repo 中的代码已在 Ubuntu 16.04 和 Python 3.6、CUDA 10.0 和 PyTorch 1.4.0 上进行了测试。 它可能适用于其他设置,但尚未经过测试。 在继续之前,请确保 CUDA 已安装并正确设置。 请注意,对于 ),CUDA 必须 >= 10.0,并且 PyTorch 和 Kaolin 安装的 CUDA 版本必须相同。 首先使用 Python 3.6 设置并激活虚拟环境并安装一些初始依赖项,例如使用 conda: conda create -n caspr_env python=3.6 conda activate caspr_env pip install -r requirements.txt 接下来安装与您的 CUDA 版本兼容的 PyTorch 1.4。 对于 CUDA 10.0,请使用以下内容(有关其他组合,请参阅 ): pip install torch==1.4.0+cu10
【文件预览】:
caspr-main
----pretrained_weights()
--------download_weights.sh(167B)
----caspr.png(1.8MB)
----requirements.txt(90B)
----LICENSE(1KB)
----caspr()
--------train.py(7KB)
--------utils()
--------models()
--------test.py(6KB)
--------data()
--------viz.py(3KB)
----README.md(13KB)
----data()
--------configs()
--------demo()
--------README.md(2KB)
--------splits()
--------download_data.sh(700B)