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'
2021-10-14 09:36:02
218KB
C++
1