From 922263418a7fadb48077b70437f3bd9e70496253 Mon Sep 17 00:00:00 2001
From: Laurent FAINSIN <laurent.fainsin@safrangroup.com>
Date: Thu, 6 Apr 2023 16:19:26 +0200
Subject: [PATCH] feat: trying function to visualize generation inside
 matplotlib

---
 test_generation.py | 15 +++++++++++++--
 1 file changed, 13 insertions(+), 2 deletions(-)

diff --git a/test_generation.py b/test_generation.py
index 9659ba0..5299b08 100644
--- a/test_generation.py
+++ b/test_generation.py
@@ -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)