pipeline for single cluster

2020-06-24 10:49:10 +02:00
parent 85cf3128f1
commit 965b805ee9
3 changed files with 6 additions and 6 deletions
--- a/main_pipeline.py
+++ b/main_pipeline.py
@ -72,7 +72,7 @@ if __name__ == '__main__':
    test_dataset = ShapeNetPartSegDataset('data', mode=GlobalVar.data_split.predict, collate_per_segment=False,
                                          refresh=True, transform=transforms)

-    grid_clusters = cluster_cubes(test_dataset[1], [3, 3, 3], max_points_per_cluster=1024)
+    grid_clusters = cluster_cubes(test_dataset[1], [1, 1, 1], max_points_per_cluster=32768)

    ps.init()