-
NeurIPS2020论文“CaSPR:学习规范时空点云表示”的emd MATLAB代码详解
资源介绍
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