1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
|
# ----------------------------------------------------------------------------
# - Open3D: www.open3d.org -
# ----------------------------------------------------------------------------
# Copyright (c) 2018-2024 www.open3d.org
# SPDX-License-Identifier: MIT
# ----------------------------------------------------------------------------
import open3d as o3d
import argparse
def preprocess_point_cloud(pcd, voxel_size):
pcd_down = pcd.voxel_down_sample(voxel_size)
pcd_down.estimate_normals(
o3d.geometry.KDTreeSearchParamHybrid(radius=voxel_size * 2.0,
max_nn=30))
pcd_fpfh = o3d.pipelines.registration.compute_fpfh_feature(
pcd_down,
o3d.geometry.KDTreeSearchParamHybrid(radius=voxel_size * 5.0,
max_nn=100))
return (pcd_down, pcd_fpfh)
if __name__ == '__main__':
pcd_data = o3d.data.DemoICPPointClouds()
parser = argparse.ArgumentParser(
'Global point cloud registration example with RANSAC')
parser.add_argument('src',
type=str,
default=pcd_data.paths[0],
nargs='?',
help='path to src point cloud')
parser.add_argument('dst',
type=str,
default=pcd_data.paths[1],
nargs='?',
help='path to dst point cloud')
parser.add_argument('--voxel_size',
type=float,
default=0.05,
help='voxel size in meter used to downsample inputs')
parser.add_argument(
'--distance_multiplier',
type=float,
default=1.5,
help='multipler used to compute distance threshold'
'between correspondences.'
'Threshold is computed by voxel_size * distance_multiplier.')
parser.add_argument('--max_iterations',
type=int,
default=64,
help='number of max FGR iterations')
parser.add_argument(
'--max_tuples',
type=int,
default=1000,
help='max number of accepted tuples for correspondence filtering')
args = parser.parse_args()
voxel_size = args.voxel_size
distance_threshold = args.distance_multiplier * voxel_size
o3d.utility.set_verbosity_level(o3d.utility.VerbosityLevel.Debug)
print('Reading inputs')
src = o3d.io.read_point_cloud(args.src)
dst = o3d.io.read_point_cloud(args.dst)
print('Downsampling inputs')
src_down, src_fpfh = preprocess_point_cloud(src, voxel_size)
dst_down, dst_fpfh = preprocess_point_cloud(dst, voxel_size)
print('Running FGR')
result = o3d.pipelines.registration.registration_fgr_based_on_feature_matching(
src_down, dst_down, src_fpfh, dst_fpfh,
o3d.pipelines.registration.FastGlobalRegistrationOption(
maximum_correspondence_distance=distance_threshold,
iteration_number=args.max_iterations,
maximum_tuple_count=args.max_tuples))
src.paint_uniform_color([1, 0, 0])
dst.paint_uniform_color([0, 1, 0])
o3d.visualization.draw([src.transform(result.transformation), dst])
|
