文件名称：ChamferDistancePytorch:Pytorch 中的倒角距离与 f-score
文件大小：19KB
文件格式：ZIP
更新时间：2024-06-18 06:58:02
Cuda
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
【文件预览】：
ChamferDistancePytorch-master
----.gitignore(19B)
----unit_test.py(2KB)
----chamfer3D()
--------setup.py(398B)
--------chamfer3D.cu(6KB)
--------chamfer_cuda.cpp(1KB)
--------dist_chamfer_3D.py(3KB)
----chamfer6D()
--------setup.py(398B)
--------chamfer_cuda.cpp(1KB)
--------chamfer6D.cu(8KB)
--------dist_chamfer_6D.py(3KB)
----LICENSE(1KB)
----chamfer2D()
--------setup.py(398B)
--------chamfer2D.cu(5KB)
--------chamfer_cuda.cpp(1KB)
--------dist_chamfer_2D.py(3KB)
----fscore.py(709B)
----README.md(3KB)
----chamfer_python.py(1KB)
----chamfer5D()
--------setup.py(398B)
--------chamfer_cuda.cpp(1KB)
--------chamfer5D.cu(7KB)
--------dist_chamfer_5D.py(3KB)