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