otthorn/nerf_plus_plus
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity

Save the raw_voxel model in pickle format

Browse source
This commit is contained in:
otthorn 2021-06-29 18:40:36 +02:00
parent ee030ce624
commit 19a8e41621
1 changed files with 5 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
voxelisation.py
View file

@ -1,4 +1,5 @@
import os import os
import pickle
import mcubes import mcubes
import numpy as np import numpy as np
@ -167,6 +168,10 @@ fg_sigma = np.array(fg_raw_voxel)
# bg_sigma = np.array(bg_raw_voxel) # bg_sigma = np.array(bg_raw_voxel)
threshold = 0.5 threshold = 0.5
# save the raw_voxel in pickle format
fd = open("raw_voxel_256.pkl", "wb")
pickel.dump(fg_sigma, fd)
fd.close()
# vertices, triangles = mcubes.marching_cubes(sigma, threshold) # vertices, triangles = mcubes.marching_cubes(sigma, threshold)
# mesh = trimesh.Trimesh(vertices / N - .5, triangles) # mesh = trimesh.Trimesh(vertices / N - .5, triangles)