Commit 1b3019a0 authored by Paul Brust's avatar Paul Brust
Browse files

improve results

parent 0d5436b9
......@@ -91,6 +91,8 @@ def do_plane_segmentation(data_directory: str, floor_name: str, results_director
for i, wall in enumerate(walls):
o3d.io.write_point_cloud(f"{results_directory}{floor_name}/result_wall_{i}.xyz", wall)
return floor_heights
def do_DBSCAN_clustering(data_directory, floor_name):
error_percentage = 0.15
......@@ -123,6 +125,8 @@ if __name__ == '__main__':
floors_directory = "./datasets/floors/"
results_directory = "./results/walls/"
floor_data = []
for floor_name in os.listdir(floors_directory):
# do_DBSCAN_clustering(floors_directory, floor_name)
......@@ -130,4 +134,9 @@ if __name__ == '__main__':
os.mkdir(f"{results_directory}{floor_name}")
except:
pass
do_plane_segmentation(floors_directory, floor_name, results_directory)
floor_data.append(do_plane_segmentation(floors_directory, floor_name, results_directory))
print(f"writing floor data")
with open(results_directory + "floor_infos.txt", "w") as f:
for floor in floor_data:
f.write(str(floor))
This diff is collapsed.
This diff is collapsed.
This diff is collapsed.
This diff is collapsed.
This diff is collapsed.
This diff is collapsed.
This diff is collapsed.
This diff is collapsed.
This diff is collapsed.
This diff is collapsed.
This diff is collapsed.
This diff is collapsed.
This diff is collapsed.
This diff is collapsed.
This diff is collapsed.
This diff is collapsed.
This diff is collapsed.
This diff is collapsed.
This diff is collapsed.
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment