3DR2N2 voxel to Mesh

3dr2n2_to_mesh.py

import numpy as np
from skimage import measure
vox = np.load('prediction.npy')

vox2 = np.pad(vox, ((1,1),(1,1),(1,1)),'constant', constant_values=0)
verts, faces, normals, values = measure.marching_cubes_lewiner(vox2, 0.0)

with open('3d.obj','w') as d:
    for v in verts:
        d.write('v {} {} {}\n'.format(v[0],v[1],v[2]))
    for f in faces:
        d.write('f {} {} {}\n'.format(f[0]+1,f[1]+1,f[2]+1))