diff --git a/curvature_analysis.py b/curvature_analysis.py index 7c875d9..062f99c 100644 --- a/curvature_analysis.py +++ b/curvature_analysis.py @@ -117,7 +117,10 @@ def generate_curvatures(v, faces, mask, neighbourhood=25, nr_of_labels=6): cloud = PyntCloud.from_file(temp_file) - k_neighbors = np.argsort(dist[mask, :][:, mask], axis=-1)[:, :neighbourhood] #cloud.get_neighbors(k=neighbourhood) + k_neighbors = np.argsort(dist[mask, :][:, mask], axis=-1)[:, :neighbourhood] + # k_neighbors = cloud.get_neighbors(k=neighbourhood) + # tree = sp.spatial.cKDTree(verts) + # k_neighbors = tree.query(verts, 25)[1] #k_neighbors = k_neighbors_cache[id] # np.argsort(dist[mask, :][:, mask], axis=-1)[:, :neighbourhood] #cloud.get_neighbors(k=neighbourhood) ev = cloud.add_scalar_field("eigen_values", k_neighbors=k_neighbors) cloud.add_scalar_field("curvature", ev=ev)