steffen/Labeling-Primitives-With-Point2Net
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

small changes

Browse Source
This commit is contained in:
Markus Friedrich 2019-09-21 22:47:08 +02:00
parent 76870a7bf8
commit 0c83d52830
1 changed files with 9 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
predict/predict.py
View File

@ -89,9 +89,15 @@ if __name__ == '__main__':
# pointcloud_file = 'C:/Projekte/csg-fitter/csg-fitter/models/0/0.xyz'
param_set = {"test1": {"cells": 2, "eps": 0.15, "min_samples": 100}}
param_set = {"test1": {"cells": 2, "eps": 0.15, "min_samples": 100},
"test2": {"cells": 2, "eps": 0.2, "min_samples": 100},
"test8": {"cells": 2, "eps": 0.1, "min_samples": 100},
"test12": {"cells": 2, "eps": 0.1, "min_samples": 100},
"test15": {"cells": 2, "eps": 0.2, "min_samples": 100}
file = "test1"
}
file = "test15"
used_params = param_set[file]
@ -241,7 +247,7 @@ if __name__ == '__main__':
sub_clusters = pc.cluster_dbscan(cluster, [0,1,2,7,8,9,10], eps=used_params["eps"], min_samples=used_params["min_samples"])
print("Sub clusters: ", len(sub_clusters))
total_clusters.extend(sub_clusters)
result_clusters = list(filter(lambda c: c.shape[0] > used_params["min_samples"], total_clusters))
for cluster in result_clusters: