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

feat: trying function to visualize generation inside matplotlib

Browse source
This commit is contained in:
Laurent FAINSIN 2023-04-06 16:19:26 +02:00
parent e8446df349
commit 922263418a
1 changed files with 13 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

15
test_generation.py
View file

@ -469,8 +469,19 @@ def generate(model, opt):
samples.append(gen)
ref.append(x)
visualize_pointcloud_batch(os.path.join(str(Path(opt.eval_path).parent), 'x.png'), gen[:64], None,
None, None)
# visualize_pointcloud_batch(
# os.path.join(str(Path(opt.eval_path).parent), 'x.png'),
# gen[:64],
# None,
# None,
# None
# )
visualize_voxels(
os.path.join(str(Path(opt.eval_path).parent), 'x.png'),
gen[:64],
1,
0.5,
)
samples = torch.cat(samples, dim=0)
ref = torch.cat(ref, dim=0)