Skip to content

GitLab

Commit 64f0ead2 authored by Jasmine's avatar Jasmine Committed by GitHub
Browse files

Merge pull request #2198 from mees/fixVisualization#2197 

Fixes broken visualization in Perspective Transformer Nets, #2197
parents 26e60496 81e6a36e
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
ptn/utils.py
View file @ 64f0ead2
...@@ -76,7 +76,7 @@ def visualize_voxel_spectral(points, vis_size=128): ...@@ -76,7 +76,7 @@ def visualize_voxel_spectral(points, vis_size=128):
points = np.rint(points) points = np.rint(points)
points = np.swapaxes(points, 0, 2) points = np.swapaxes(points, 0, 2)
fig = p.figure(figsize=(1, 1), dpi=vis_size) fig = p.figure(figsize=(1, 1), dpi=vis_size)
verts, faces = measure.marching_cubes(points, 0, spacing=(0.1, 0.1, 0.1)) verts, faces = measure.marching_cubes_classic(points, 0, spacing=(0.1, 0.1, 0.1))
ax = fig.add_subplot(111, projection='3d') ax = fig.add_subplot(111, projection='3d')
ax.plot_trisurf( ax.plot_trisurf(
verts[:, 0], verts[:, 1], faces, verts[:, 2], cmap='Spectral_r', lw=0.1) verts[:, 0], verts[:, 1], faces, verts[:, 2], cmap='Spectral_r', lw=0.1)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment