requirements updated

main_pipeline.py

@@ -72,7 +72,9 @@ if __name__ == '__main__':
     test_dataset = ShapeNetPartSegDataset('data', mode=GlobalVar.data_split.predict, collate_per_segment=False,
                                           refresh=True, transform=transforms)  # , cluster_type='pc')
 
-    grid_clusters = cluster_cubes(test_dataset[0], [1, 1, 1], max_points_per_cluster=8192)
+
+    # Here you can choose which dataset to load, in case there are more than one pointcloud files in raw/predict
+    grid_clusters = cluster_cubes(test_dataset[0], [1, 1, 1], max_points_per_cluster=4096)
 
     ps.init()