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,
# None # None
# ) # )
visualize_voxels( # visualize_voxels(
os.path.join(str(Path(opt.eval_path).parent), 'x.png'), # os.path.join(str(Path(opt.eval_path).parent), 'x.png'),
gen[:64], # gen[:64],
1, # 1,
0.5, # 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) samples = torch.cat(samples, dim=0)
ref = torch.cat(ref, dim=0) ref = torch.cat(ref, dim=0)