pytorch3d

pytorch3d copied to clipboard

🚀 Feature

Support bounding volume hierarchy (BVH) for distance computation

Motivation

Minimal distance computation is important for 3D vision and graphics tasks, e.g. signed distance field computation, soft body simulation, and ray tracing. Bounding volume hierarchy can accelerate this process greatly by putting the geometric objects into a tree structure, and it can be supported by GPU.

I think it would be great if PyTorch3D can add the bounding volume hierarchy structure.

Pitch

There is a GitHub repository that supports the BVH structure with PyTorch wrapper https://github.com/vchoutas/torch-mesh-isect. I find the current point_mesh_distance implementation is done by traveling through all points and faces in the point cloud and mesh, so I think it would be good to use BVH. I am glad to give a suggestion and thanks for your contribution!