mcubes_pytorch:PyTorch环境的Marching cubes实现-源码

PyTorch中的行进多维数据集 用于PyTorch环境的行进多维数据集。 后端是使用C ++和CUDA实现的。 当前，CUDA代码仅适用于具有2的幂的大小的网格单元（例如32、64、128，...）。 如果您希望使用非2的幂的代码，请在使用零之前用零填充以使尺寸成为2的幂。 有关详情，请参见 。 建造 $ python setup.py build_ext -i 用法 有关详细信息，请参见 。 import numpy as np import open3d as o3d import torch from mcubes import marching_cubes , grid_interp # Grid data N = 128 x , y , z = np . mgrid [: N , : N , : N ] x = ( x / N ). astype ( 'float32'