• Nikhila Ravi's avatar
    Classic Marching Cubes algorithm implementation · ebac66da
    Nikhila Ravi authored
    Summary:
    Defines a function to run marching cubes algorithm on a single or batch of 3D scalar fields. Returns a mesh's faces and vertices.
    
    UPDATES (12/18)
    - Input data is now specified as a (B, D, H, W) tensor as opposed to a (B, W, H, D) tensor. This will now be compatible with the Volumes datastructure.
    - Add an option to return output vertices in local coordinates instead of world coordinates.
    Also added a small fix to remove the dype for device in Transforms3D - if passing in a torch.device instead of str it causes a pyre error.
    
    Reviewed By: jcjohnson
    
    Differential Revision: D24599019
    
    fbshipit-source-id: 90554a200319fed8736a12371cc349e7108aacd0
    ebac66da
test_marching_cubes.py 28.1 KB