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 ==========================