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

Added normals to prediction DataObject

Browse Source
This commit is contained in:
Si11ium
2019-08-07 08:54:07 +02:00
parent efba70f19a
commit 74de208831
4 changed files with 20 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
predict/pointclouds/0_pc.xyz
View File

@@ -1,4 +1,5 @@
0.0759953 -0.395683 -0.130632 -0.871549 -0.481571 0.0921502
2049 6
0.0759953 -0.395683 -0.130632 -0.871549 -0.481571 0.0921502
-0.364005 0.664317 0.169368 -0.997143 0.00181983 -0.0755164
0.415995 0.424317 -0.150632 0.920812 -0.223131 -0.319865
-0.364005 0.104317 -0.290632 -0.678055 0.734933 -0.0107393

10
predict/predict.py
View File

@@ -260,9 +260,9 @@ if __name__ == '__main__':
dataset_folder = './data/raw/predict/'
pointcloud_file = './pointclouds/0_pc.xyz'
pc = pc.read_pointcloud(pointcloud_file)
pc = normalize_pointcloud(pc)
pc = append_normal_angles(pc)
pcloud = pc.read_pointcloud(pointcloud_file)
pcloud = normalize_pointcloud(pcloud)
pcloud = append_normal_angles(pcloud)
# pc = StandardScaler().fit_transform(pc)
@@ -270,10 +270,10 @@ if __name__ == '__main__':
# Add full point cloud to prediction folder.
recreate_folder(dataset_folder + '0_0' + '/')
pc_fps = farthest_point_sampling(pc, opt.npoints)
pc_fps = farthest_point_sampling(pcloud, opt.npoints)
pc.write_pointcloud(dataset_folder + '0_0' + '/pc.xyz', pc_fps)
pc_clusters = extract_cube_clusters(pc, [4,4,4], 1024)
pc_clusters = extract_cube_clusters(pcloud, [4,4,4], 1024)
#pc_clusters = extract_clusters(pc, [0, 1, 2, 3, 4, 5], eps=0.1, min_samples=0.0001, metric='euclidean', algo='auto')
# Add cluster point clouds to prediction folder.
for idx, pcc in enumerate(pc_clusters):