-
在PyTorch环境下实现Marching cubes的mcubes_pytorch
资源介绍
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'