From 772c5594b70e274dd14acaa83348ef3a4249e2dc Mon Sep 17 00:00:00 2001
From: Markus Friedrich <markus.friedrich@ifi.lmu.de>
Date: Sun, 28 Jun 2020 17:49:40 +0200
Subject: [PATCH] full pipeline is now in place

---
 main_pipeline.py | 9 ++++-----
 1 file changed, 4 insertions(+), 5 deletions(-)

diff --git a/main_pipeline.py b/main_pipeline.py
index 8d6a71b..dab5a7e 100644
--- a/main_pipeline.py
+++ b/main_pipeline.py
@@ -60,11 +60,11 @@ def predict_prim_type(input_pc, model):
 
 if __name__ == '__main__':
 
-    display_mode = DisplayMode.Clusters
-    grid_cluster_max_pts = 3000#8192
-    grid_clusters = [3,3,3]
+    display_mode = DisplayMode.Types
+    grid_cluster_max_pts = 32 * 1024
+    grid_clusters = [1, 1, 1]
     type_cluster_eps = 0.1
-    type_cluster_min_pts = 50
+    type_cluster_min_pts = 100
 
     model_path = Path('output') / 'PN2' / 'PN_9843bf499399786cfd58fe79fa1b3db8' / 'version_0'
 
@@ -77,7 +77,6 @@ if __name__ == '__main__':
 
     grid_clusters = cluster_cubes(test_dataset[0], grid_clusters, max_points_per_cluster=grid_cluster_max_pts)
 
-
     ps.init()
 
     # ========================== Grid Clustering ==========================