Skip to content

Commit 1468a53

Browse files
committed
pykdtree added and metrics
1 parent c9a8310 commit 1468a53

File tree

2 files changed

+20
-0
lines changed

2 files changed

+20
-0
lines changed

src/metrics.py

+19
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,19 @@
1+
from pykdtree.kdtree import KDTree
2+
import numpy as np
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']

src/pykdtree

+1
Original file line numberDiff line numberDiff line change
@@ -0,0 +1 @@
1+
Subproject commit 1aa176a580c865d649d11f28384cd32420cc3c31

0 commit comments

Comments
 (0)