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
|
// ----------------------------------------------------------------------------
// - Open3D: www.open3d.org -
// ----------------------------------------------------------------------------
// Copyright (c) 2018-2024 www.open3d.org
// SPDX-License-Identifier: MIT
// ----------------------------------------------------------------------------
#include <Eigen/Dense>
#include <iostream>
#include <memory>
#include "open3d/Open3D.h"
using namespace open3d;
void VisualizeRegistration(const open3d::geometry::PointCloud &source,
const open3d::geometry::PointCloud &target,
const Eigen::Matrix4d &Transformation) {
std::shared_ptr<geometry::PointCloud> source_transformed_ptr(
new geometry::PointCloud);
std::shared_ptr<geometry::PointCloud> target_ptr(new geometry::PointCloud);
*source_transformed_ptr = source;
*target_ptr = target;
source_transformed_ptr->Transform(Transformation);
visualization::DrawGeometries({source_transformed_ptr, target_ptr},
"Registration result");
}
void PrintHelp() {
using namespace open3d;
PrintOpen3DVersion();
// clang-format off
utility::LogInfo("Usage:");
utility::LogInfo(" > RegistrationColoredICP source_pcd target_pcd [--visualize]");
// clang-format on
utility::LogInfo("");
}
int main(int argc, char *argv[]) {
using namespace open3d;
utility::SetVerbosityLevel(utility::VerbosityLevel::Debug);
if (argc < 3 ||
utility::ProgramOptionExistsAny(argc, argv, {"-h", "--help"})) {
PrintHelp();
return 1;
}
bool visualize = false;
if (utility::ProgramOptionExists(argc, argv, "--visualize")) {
visualize = true;
}
// Prepare input
std::shared_ptr<geometry::PointCloud> source =
open3d::io::CreatePointCloudFromFile(argv[1]);
std::shared_ptr<geometry::PointCloud> target =
open3d::io::CreatePointCloudFromFile(argv[2]);
if (source == nullptr || target == nullptr) {
utility::LogWarning("Unable to load source or target file.");
return -1;
}
std::vector<double> voxel_sizes = {0.05, 0.05 / 2, 0.05 / 4};
std::vector<int> iterations = {50, 30, 14};
Eigen::Matrix4d trans = Eigen::Matrix4d::Identity();
for (int i = 0; i < 3; ++i) {
float voxel_size = voxel_sizes[i];
auto source_down = source->VoxelDownSample(voxel_size);
source_down->EstimateNormals(open3d::geometry::KDTreeSearchParamHybrid(
voxel_size * 2.0, 30));
auto target_down = target->VoxelDownSample(voxel_size);
target_down->EstimateNormals(open3d::geometry::KDTreeSearchParamHybrid(
voxel_size * 2.0, 30));
auto result = pipelines::registration::RegistrationColoredICP(
*source_down, *target_down, 0.07, trans,
pipelines::registration::
TransformationEstimationForColoredICP(),
pipelines::registration::ICPConvergenceCriteria(1e-6, 1e-6,
iterations[i]));
trans = result.transformation_;
if (visualize) {
VisualizeRegistration(*source, *target, trans);
}
}
std::stringstream ss;
ss << trans;
utility::LogInfo("Final transformation = \n{}", ss.str());
return 0;
}
|
