feat: trying function to visualize generation inside matplotlib

This commit is contained in:
Laurent FAINSIN 2023-04-06 16:19:26 +02:00
parent e8446df349
commit 922263418a

View file

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