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
103
104
|
// Copyright (c) 2020, Viktor Larsson
// All rights reserved.
//
// Redistribution and use in source and binary forms, with or without
// modification, are permitted provided that the following conditions are met:
//
// * Redistributions of source code must retain the above copyright
// notice, this list of conditions and the following disclaimer.
//
// * Redistributions in binary form must reproduce the above copyright
// notice, this list of conditions and the following disclaimer in the
// documentation and/or other materials provided with the distribution.
//
// * Neither the name of the copyright holder nor the
// names of its contributors may be used to endorse or promote products
// derived from this software without specific prior written permission.
//
// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
// AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
// IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
// ARE DISCLAIMED. IN NO EVENT SHALL <COPYRIGHT HOLDER> BE LIABLE FOR ANY
// DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
// (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
// LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
// ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
// (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
// SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
#include "up2p.h"
#include "PoseLib/misc/univariate.h"
int poselib::up2p(const std::vector<Eigen::Vector3d> &x, const std::vector<Eigen::Vector3d> &X,
poselib::CameraPoseVector *output) {
Eigen::Matrix<double, 4, 4> A;
Eigen::Matrix<double, 4, 2> b;
A << -x[0](2), 0, x[0](0), X[0](0) * x[0](2) - X[0](2) * x[0](0), 0, -x[0](2), x[0](1),
-X[0](1) * x[0](2) - X[0](2) * x[0](1), -x[1](2), 0, x[1](0), X[1](0) * x[1](2) - X[1](2) * x[1](0), 0,
-x[1](2), x[1](1), -X[1](1) * x[1](2) - X[1](2) * x[1](1);
b << -2 * X[0](0) * x[0](0) - 2 * X[0](2) * x[0](2), X[0](2) * x[0](0) - X[0](0) * x[0](2), -2 * X[0](0) * x[0](1),
X[0](2) * x[0](1) - X[0](1) * x[0](2), -2 * X[1](0) * x[1](0) - 2 * X[1](2) * x[1](2),
X[1](2) * x[1](0) - X[1](0) * x[1](2), -2 * X[1](0) * x[1](1), X[1](2) * x[1](1) - X[1](1) * x[1](2);
// b = A.partialPivLu().solve(b);
b = A.inverse() * b;
const double c2 = b(3, 0);
const double c3 = b(3, 1);
double qq[2];
const int sols = univariate::solve_quadratic_real(1.0, c2, c3, qq);
output->clear();
for (int i = 0; i < sols; ++i) {
const double q = qq[i];
const double q2 = q * q;
const double inv_norm = 1.0 / (1 + q2);
const double cq = (1 - q2) * inv_norm;
const double sq = 2 * q * inv_norm;
Eigen::Matrix3d R;
R.setIdentity();
R(0, 0) = cq;
R(0, 2) = sq;
R(2, 0) = -sq;
R(2, 2) = cq;
Eigen::Vector3d t;
t = b.block<3, 1>(0, 0) * q + b.block<3, 1>(0, 1);
t *= -inv_norm;
output->emplace_back(R, t);
}
return sols;
}
int poselib::up2p(const std::vector<Eigen::Vector3d> &x, const std::vector<Eigen::Vector3d> &X,
const Eigen::Vector3d &g_cam, const Eigen::Vector3d &g_world, CameraPoseVector *output) {
// Rotate camera world coordinate system
Eigen::Matrix3d Rc = Eigen::Quaterniond::FromTwoVectors(g_cam, Eigen::Vector3d::UnitY()).toRotationMatrix();
Eigen::Matrix3d Rw = Eigen::Quaterniond::FromTwoVectors(g_world, Eigen::Vector3d::UnitY()).toRotationMatrix();
std::vector<Eigen::Vector3d> x_upright = x;
std::vector<Eigen::Vector3d> X_upright = X;
for (int i = 0; i < 2; ++i) {
x_upright[i] = Rc * x[i];
X_upright[i] = Rw * X[i];
}
int n_sols = up2p(x_upright, X_upright, output);
// De-rotate coordinate systems
for (int i = 0; i < n_sols; ++i) {
Eigen::Matrix3d R = (*output)[i].R();
Eigen::Vector3d t = (*output)[i].t;
t = Rc.transpose() * t;
R = Rc.transpose() * R * Rw;
(*output)[i] = CameraPose(R, t);
}
return n_sols;
}
|
