Safran/PVD
1
0
Fork 0
Code Issues Pull requests Packages Projects Releases Wiki Activity

feat: ghetto matplotlib viz

Browse source
This commit is contained in:
Laurent FAINSIN 2023-04-11 09:16:29 +02:00
parent 922263418a
commit 52336ac82c
1 changed files with 25 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

31
test_generation.py
View file

@ -476,12 +476,31 @@ def generate(model, opt):
# None,
# None
# )
visualize_voxels(
os.path.join(str(Path(opt.eval_path).parent), 'x.png'),
gen[:64],
1,
0.5,
)
# visualize_voxels(
# os.path.join(str(Path(opt.eval_path).parent), 'x.png'),
# gen[:64],
# 1,
# 0.5,
# )
# visualize using matplotlib
import matplotlib.pyplot as plt
import matplotlib.cm as cm
import matplotlib
matplotlib.use('TkAgg')
for idx, pc in enumerate(gen[:64]):
print(f"Visualizing point cloud {idx}...")
fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')
ax.scatter(pc[:,0], pc[:,1], pc[:,2], c=pc[:,2], cmap=cm.jet)
ax.set_aspect('equal')
ax.axis('off')
# ax.set_xlim(-1, 1)
# ax.set_ylim(-1, 1)
# ax.set_zlim(-1, 1)
plt.show()
plt.close()
samples = torch.cat(samples, dim=0)
ref = torch.cat(ref, dim=0)