steffen/point_to_primitive
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

full pipeline is now in place

Browse Source
This commit is contained in:
Markus Friedrich 2020-06-28 17:49:40 +02:00
parent 84c879e5bf
commit 772c5594b7
1 changed files with 4 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
main_pipeline.py
View File

@ -60,11 +60,11 @@ def predict_prim_type(input_pc, model):
if __name__ == '__main__': if __name__ == '__main__':
display_mode = DisplayMode.Clusters display_mode = DisplayMode.Types
grid_cluster_max_pts = 3000#8192 grid_cluster_max_pts = 32 * 1024
grid_clusters = [3,3,3] grid_clusters = [1, 1, 1]
type_cluster_eps = 0.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' 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) grid_clusters = cluster_cubes(test_dataset[0], grid_clusters, max_points_per_cluster=grid_cluster_max_pts)
ps.init() ps.init()
# ========================== Grid Clustering ========================== # ========================== Grid Clustering ==========================