mcubes_pytorch:PyTorch环境的Marching cubes实现

时间:2021-05-19 05:48:48
【文件属性】:
文件名称:mcubes_pytorch:PyTorch环境的Marching cubes实现
文件大小:218KB
文件格式:ZIP
更新时间:2021-05-19 05:48:48
C++ 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'
【文件预览】:
mcubes_pytorch-master
----mcubes.py(3KB)
----metaball.png(195KB)
----cxx()
--------helper_math.h(36KB)
--------grid_interp_cpu.cpp(2KB)
--------mcubes.cpp(881B)
--------mcubes_cuda.cu(27KB)
--------pscan.cu(5KB)
--------cuda_utils.h(855B)
--------macros.h(2KB)
--------mcubes_cpu.cpp(26KB)
--------grid_interp_cuda.cu(4KB)
--------pscan.h(98B)
----setup.py(1KB)
----README.md(2KB)
----.gitignore(32B)
----environment.yaml(192B)

网友评论