1
1
from pykdtree .kdtree import KDTree
2
2
import numpy as np
3
3
4
- pc_path1 = '/home/madhvi/Documents/CV Project/data/subset/ShapeNet/02691156/1ac29674746a0fc6b87697d3904b168b/pointcloud.npz'
5
- pc_path2 = '/home/madhvi/Documents/CV Project/data/subset/ShapeNet/02691156/1ac29674746a0fc6b87697d3904b168b/pointcloud.npz'
6
- point_path1 = '/home/madhvi/Documents/CV Project/data/subset/ShapeNet/02691156/1ac29674746a0fc6b87697d3904b168b/points.npz'
7
- point_path2 = '/home/madhvi/Documents/CV Project/data/subset/ShapeNet/02691156/1ac29674746a0fc6b87697d3904b168b/points.npz'
8
-
9
- pc_data1 = np .load (pc_path1 )
10
- pc_data2 = np .load (pc_path2 )
11
- points_data1 = np .load (point_path1 )
12
- points_data2 = np .load (point_path2 )
13
-
14
- pointcloud = pc_data1 ['points' ]
15
- pointcloud_gt = pc_data2 ['points' ]
16
- normals = pc_data1 ['normals' ]
17
- normals_gt = pc_data2 ['normals' ]
18
- occ_1 = points_data1 ['occupancies' ]
19
- occ_2 = points_data2 ['occupancies' ]
20
-
21
4
22
5
def compute_iou (occ1 , occ2 ):
23
6
''' Computes the Intersection over Union (IoU) value for two sets of
@@ -155,5 +138,22 @@ def eval_pointcloud(pointcloud, pointcloud_gt,
155
138
return out_dict
156
139
157
140
if __name__ == '__main__' :
141
+ pc_path1 = '/home/madhvi/Documents/CV Project/data/subset/ShapeNet/02691156/1ac29674746a0fc6b87697d3904b168b/pointcloud.npz'
142
+ pc_path2 = '/home/madhvi/Documents/CV Project/data/subset/ShapeNet/02691156/1ac29674746a0fc6b87697d3904b168b/pointcloud.npz'
143
+ point_path1 = '/home/madhvi/Documents/CV Project/data/subset/ShapeNet/02691156/1ac29674746a0fc6b87697d3904b168b/points.npz'
144
+ point_path2 = '/home/madhvi/Documents/CV Project/data/subset/ShapeNet/02691156/1ac29674746a0fc6b87697d3904b168b/points.npz'
145
+
146
+ pc_data1 = np .load (pc_path1 )
147
+ pc_data2 = np .load (pc_path2 )
148
+ points_data1 = np .load (point_path1 )
149
+ points_data2 = np .load (point_path2 )
150
+
151
+ pointcloud = pc_data1 ['points' ]
152
+ pointcloud_gt = pc_data2 ['points' ]
153
+ normals = pc_data1 ['normals' ]
154
+ normals_gt = pc_data2 ['normals' ]
155
+ occ_1 = points_data1 ['occupancies' ]
156
+ occ_2 = points_data2 ['occupancies' ]
157
+
158
158
eval_dict = eval_pointcloud (pointcloud , pointcloud_gt , normals , normals_gt , occ_1 , occ_2 )
159
- print (eval_dict )
159
+ print (eval_dict )
0 commit comments