倒角距离API
倒角距离（CD）是两个点云的评估指标。 它考虑了每个点的距离。 对于每个云中的每个点，CD都会找到另一个点集中的最近点，并将距离的平方求和。 在Shapenet的形状重构挑战中使用了它。
点云S1和S2之间的倒角距离定义为
本自述文件是有关如何编译倒角距离的代码的指南，以及使用倒角距离api的教程。
先决条件
5.4.0
9.0
2.7.12
1.7.0
编译倒角距离代码
文件夹chamfer-distance包含用于倒角距离的tensorflow模块。 要编译它们，请确保已安装tensorflow。 然后，根据您的环境修改chamfer-distance/Makefile的前4行。 最后，编译倒角距离代码
# From chamfer-distance/
make
要测试代码，请尝试
# From chamfer-distance/
python t
2021-09-06 14:17:22
10KB
1
pip install torch ninja
Pytorch 倒角距离。
包括一个CUDA版本和一个带有 pytorch 标准操作的PYTHON版本。 注意：在这个仓库中，dist1 和 dist2 是点云欧氏距离的平方，因此您应该相应地调整阈值。
F - 分数
CUDA 版本
即时编译
支持多GPU
二维点云。
3D 点云。
5D 点云。
连续（）安全。
Python版本
支持任意维度
用法
import torch , chamfer3D . dist_chamfer_3D , fscore
chamLoss = chamfer3D . dist_chamfer_3D . chamfer_3DDist ()
points1 = torch . rand ( 32 , 1000 , 3 ). cuda ()
points2 = torch . rand ( 32
2021-08-16 16:14:56
19KB
1