Skip to content

Commit 7707d58

Browse files
committed
Fix conflicts
2 parents 544c9d2 + b661526 commit 7707d58

File tree

1 file changed

+18
-18
lines changed

1 file changed

+18
-18
lines changed

src/metrics.py

Lines changed: 18 additions & 18 deletions
Original file line numberDiff line numberDiff line change
@@ -1,23 +1,6 @@
11
from pykdtree.kdtree import KDTree
22
import numpy as np
33

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-
214

225
def compute_iou(occ1, occ2):
236
''' Computes the Intersection over Union (IoU) value for two sets of
@@ -155,5 +138,22 @@ def eval_pointcloud(pointcloud, pointcloud_gt,
155138
return out_dict
156139

157140
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+
158158
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

Comments
 (0)