本文整理汇总了Python中scipy.spatial.KDTree方法的典型用法代码示例。如果您正苦于以下问题:Python spatial.KDTree方法的具体用法?Python spatial.KDTree怎么用?Python spatial.KDTree使用的例子?那么恭喜您, 这里精选的方法代码示例或许可以为您提供帮助。您也可以进一步了解该方法所在模块scipy.spatial的用法示例。
在下文中一共展示了spatial.KDTree方法的23个代码示例,这些例子默认根据受欢迎程度排序。您可以为喜欢或者感觉有用的代码点赞,您的评价将有助于我们的系统推荐出更棒的Python代码示例。
示例1: generate_icp_results
点赞 6
# 需要导入模块: from scipy import spatial [as 别名]
# 或者: from scipy.spatial import KDTree [as 别名]
def generate_icp_results(self, gt_pose):
from icp import icp_test
from scipy.spatial import KDTree
M_given = self.templates[self.template_idx,:,:]
S_given = helper.apply_transformation(M_given.reshape((1,-1,3)), gt_pose)[0]
# S_given = S_given + np.random.normal(0,1,S_given.shape)# Noisy Data
M_given = M_given[0:self.NUM_POINT,:]# template data
S_given = S_given[0:self.NUM_POINT,:]# source data
tree_M = KDTree(M_given)
tree_M_sampled = KDTree(M_given[0:100,:])
final_pose, model_data, sensor_data, predicted_data, title, _, _ = icp_test(S_given[0:100,:], M_given, tree_M, M_given[0:100,:], tree_M_sampled, S_given, gt_pose.reshape((1,6)), self.MAX_LOOPS, self.ftol)
self.find_errors(gt_pose, final_pose)
helper.display_three_clouds(model_data, sensor_data, predicted_data, title)
开发者ID:vinits5,项目名称:pointnet-registration-framework,代码行数:19,
示例2: bench_build
点赞 6
# 需要导入模块: from scipy import spatial [as 别名]
# 或者: from scipy.spatial import KDTree [as 别名]
def bench_build(self):
print()
print(' Constructing kd-tree')
print('=====================================')
print(' dim | # points | KDTree | cKDTree ')
for (m, n, repeat) in [(3,10000,3), (8,10000,3), (16,10000,3)]:
print('%4s | %7s ' % (m, n), end=' ')
sys.stdout.flush()
data = np.concatenate((np.random.randn(n//2,m),
np.random.randn(n-n//2,m)+np.ones(m)))
print('| %6.3fs ' % (measure('T1 = KDTree(data)', repeat) / repeat), end=' ')
sys.stdout.flush()
print('| %6.3fs' % (measure('T2 = cKDTree(data)', repeat) / repeat), end=' ')
sys.stdout.flush()
print('')
开发者ID:ktraunmueller,项目名称:Computable,代码行数:20,
示例3: setUp
点赞 6
# 需要导入模块: from scipy import spatial [as 别名]
# 或者: from scipy.spatial import KDTree [as 别名]
def setUp(self):
self.data = np.array([[0,0,0],
[0,0,1],
[0,1,0],
[0,1,1],
[1,0,0],
[1,0,1],
[1,1,0],
[1,1,1]])
self.kdtree = KDTree(self.data)
self.n = self.kdtree.n
self.m = self.kdtree.m
np.random.seed(1234)
self.x = np.random.randn(3)
self.d = 0.5
self.k = 4
开发者ID:ktraunmueller,项目名称:Computable,代码行数:18,
示例4: test_onetree_query
点赞 6
# 需要导入模块: from scipy import spatial [as 别名]
# 或者: from scipy.spatial import KDTree [as 别名]
def test_onetree_query():
np.random.seed(0)
n = 50
k = 4
points = np.random.randn(n,k)
T = KDTree(points)
yield check_onetree_query, T, 0.1
points = np.random.randn(3*n,k)
points[:n] *= 0.001
points[n:2*n] += 2
T = KDTree(points)
yield check_onetree_query, T, 0.1
yield check_onetree_query, T, 0.001
yield check_onetree_query, T, 0.00001
yield check_onetree_query, T, 1e-6
开发者ID:ktraunmueller,项目名称:Computable,代码行数:18,
示例5: exact_full_marker_data
点赞 6
# 需要导入模块: from scipy import spatial [as 别名]
# 或者: from scipy.spatial import KDTree [as 别名]
def exact_full_marker_data(csv_path, marker_pkl):
all_data_arr = np.genfromtxt(csv_path, delimiter=",", skip_header=1)
marker_jdcs_collection = cPickle.load(open(marker_pkl[0], "rb"))
tmp_list = list()
for jdc in marker_jdcs_collection:
tmp_list.extend(jdc)
marker_pcd_arr = np.array(tmp_list)
tree = spatial.KDTree(all_data_arr[:, :3])
marker_full_data_ls = []
for i in xrange(marker_pcd_arr.shape[0]):
ret = tree.query(marker_pcd_arr[i])
marker_full_data_ls.append(all_data_arr[ret[1]])
marker_full_data_arr = np.array(marker_full_data_ls)
return marker_full_data_arr
# get the Hessian normal form of 3D plane
开发者ID:mfxox,项目名称:ILCC,代码行数:21,
示例6: build_index
点赞 6
# 需要导入模块: from scipy import spatial [as 别名]
# 或者: from scipy.spatial import KDTree [as 别名]
def build_index(self):
midpoints = []
self.index = {}
for ct, fset in enumerate(self.sets.values()):
mp = []
for vr in self.explanatory_variables:
mp.append(fset.sets[vr.name].centroid)
midpoints.append(mp)
self.index[ct] = fset.name
import sys
sys.setrecursionlimit(100000)
self.kdtree = KDTree(midpoints)
sys.setrecursionlimit(1000)
开发者ID:PYFTS,项目名称:pyFTS,代码行数:21,
示例7: build_index
点赞 6
# 需要导入模块: from scipy import spatial [as 别名]
# 或者: from scipy.spatial