Merge branch 'master' of https://gitlab.lrz.de/mobile-ifi/point_to_primitive

# Conflicts: # main_pipeline.py
2020-06-26 16:32:19 +02:00
parent fc88c687ae cf99ac1aec
commit 84c879e5bf
4 changed files with 7 additions and 5 deletions
--- a/main_pipeline.py
+++ b/main_pipeline.py
@ -67,15 +67,17 @@ if __name__ == '__main__':
    type_cluster_min_pts = 50

    model_path = Path('output') / 'PN2' / 'PN_9843bf499399786cfd58fe79fa1b3db8' / 'version_0'
+
    loaded_model = restore_logger_and_model(model_path)
    loaded_model.eval()

    transforms = Compose([NormalizeScale(), ])
    test_dataset = ShapeNetPartSegDataset('data', mode=GlobalVar.data_split.predict, collate_per_segment=False,
-                                          refresh=True, transform=transforms)
+                                          refresh=True, transform=transforms)  # , cluster_type='pc')

    grid_clusters = cluster_cubes(test_dataset[0], grid_clusters, max_points_per_cluster=grid_cluster_max_pts)

+
    ps.init()

    # ========================== Grid Clustering ==========================