diff --git a/predict/predict.py b/predict/predict.py index 59e18ba..bd83712 100644 --- a/predict/predict.py +++ b/predict/predict.py @@ -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: