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
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
|
# ----------------------------------------------------------------------------
# - Open3D: www.open3d.org -
# ----------------------------------------------------------------------------
# Copyright (c) 2018-2024 www.open3d.org
# SPDX-License-Identifier: MIT
# ----------------------------------------------------------------------------
# examples/python/reconstruction_system/slac.py
import numpy as np
import open3d as o3d
import os, sys
pyexample_path = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
sys.path.append(pyexample_path)
from open3d_example import join, get_file_list, write_poses_to_log
sys.path.append(os.path.dirname(os.path.abspath(__file__)))
def run(config):
print("slac non-rigid optimization.")
o3d.utility.set_verbosity_level(o3d.utility.VerbosityLevel.Debug)
path_dataset = config['path_dataset']
ply_file_names = get_file_list(
join(config["path_dataset"], config["folder_fragment"]), ".ply")
if (len(ply_file_names) == 0):
raise RuntimeError(
"No fragment found in {}, please make sure the reconstruction_system has finished running on the dataset."
.format(join(config["path_dataset"], config["folder_fragment"])))
pose_graph_fragment = o3d.io.read_pose_graph(
join(path_dataset, config["template_refined_posegraph_optimized"]))
# SLAC optimizer parameters.
slac_params = o3d.t.pipelines.slac.slac_optimizer_params(
max_iterations=config["max_iterations"],
voxel_size=config["voxel_size"],
distance_threshold=config["distance_threshold"],
fitness_threshold=config["fitness_threshold"],
regularizer_weight=config["regularizer_weight"],
device=o3d.core.Device(str(config["device"])),
slac_folder=join(path_dataset, config["folder_slac"]))
# SLAC debug option.
debug_option = o3d.t.pipelines.slac.slac_debug_option(False, 0)
# Run the system.
pose_graph_updated = o3d.pipelines.registration.PoseGraph()
# rigid optimization method.
if (config["method"] == "rigid"):
pose_graph_updated = o3d.t.pipelines.slac.run_rigid_optimizer_for_fragments(
ply_file_names, pose_graph_fragment, slac_params, debug_option)
elif (config["method"] == "slac"):
pose_graph_updated, ctrl_grid = o3d.t.pipelines.slac.run_slac_optimizer_for_fragments(
ply_file_names, pose_graph_fragment, slac_params, debug_option)
hashmap = ctrl_grid.get_hashmap()
active_buf_indices = hashmap.active_buf_indices().to(
o3d.core.Dtype.Int64)
key_tensor = hashmap.key_tensor()[active_buf_indices]
key_tensor.save(
join(slac_params.get_subfolder_name(), "ctr_grid_keys.npy"))
value_tensor = hashmap.value_tensor()[active_buf_indices]
value_tensor.save(
join(slac_params.get_subfolder_name(), "ctr_grid_values.npy"))
else:
raise RuntimeError(
"Requested optimization method {}, is not implemented. Implemented methods includes slac and rigid."
.format(config["method"]))
# Write updated pose graph.
o3d.io.write_pose_graph(
join(slac_params.get_subfolder_name(),
config["template_optimized_posegraph_slac"]), pose_graph_updated)
# Write trajectory for slac-integrate stage.
fragment_folder = join(path_dataset, config["folder_fragment"])
params = []
for i in range(len(pose_graph_updated.nodes)):
fragment_pose_graph = o3d.io.read_pose_graph(
join(fragment_folder, "fragment_optimized_%03d.json" % i))
for node in fragment_pose_graph.nodes:
pose = np.dot(pose_graph_updated.nodes[i].pose, node.pose)
param = o3d.camera.PinholeCameraParameters()
param.extrinsic = np.linalg.inv(pose)
params.append(param)
trajectory = o3d.camera.PinholeCameraTrajectory()
trajectory.parameters = params
o3d.io.write_pinhole_camera_trajectory(
slac_params.get_subfolder_name() + "/optimized_trajectory_" +
str(config["method"]) + ".log", trajectory)
|
