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
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
|
#!/usr/bin/env python3
r'''diff test
Make sure the projection-diff function produces correct results
'''
import sys
import numpy as np
import numpysane as nps
import os
testdir = os.path.dirname(os.path.realpath(__file__))
# I import the LOCAL mrcal since that's what I'm testing
sys.path[:0] = f"{testdir}/..",
import mrcal
import testutils
model_opencv8 = mrcal.cameramodel(f"{testdir}/data/cam0.opencv8.cameramodel")
model_splined = mrcal.cameramodel(f"{testdir}/data/cam0.splined.cameramodel")
gridn_width = 50
########## Compare the model to itself. I should get 0 diff and identity transform
difflen, diff, q0, implied_Rt10 = \
mrcal.projection_diff( (model_splined,model_splined),
gridn_width = gridn_width,
distance = None,
use_uncertainties = False )
testutils.confirm_equal( difflen.shape[1], gridn_width,
msg = "Expected number of columns" )
testutils.confirm_equal( difflen.shape[0], int(round( model_splined.imagersize()[1] / model_splined.imagersize()[0] * gridn_width)),
msg = "Expected number of rows" )
icenter = np.array(difflen.shape) // 2
testutils.confirm_equal( difflen*0, difflen,
eps = 0.08,
worstcase = True,
relative = False,
msg = "diff(model,model) at infinity should be 0")
testutils.confirm_equal( 0, np.arccos((np.trace(implied_Rt10[:3,:]) - 1) / 2.) * 180./np.pi,
eps = 0.01,
msg = "diff(model,model) at infinity should produce a rotation of 0 deg")
testutils.confirm_equal( 0, nps.mag(implied_Rt10[3,:]),
eps = 0.01,
msg = "diff(model,model) at infinity should produce a translation of 0 m")
difflen, diff, q0, implied_Rt10 = \
mrcal.projection_diff( (model_splined,model_splined),
gridn_width = 50,
distance = 3.,
use_uncertainties = False )
testutils.confirm_equal( difflen*0, difflen,
eps = 0.08,
worstcase = True,
relative = False,
msg = "diff(model,model) at 3m should be 0")
testutils.confirm_equal( 0, np.arccos((np.trace(implied_Rt10[:3,:]) - 1) / 2.) * 180./np.pi,
eps = 0.01,
msg = "diff(model,model) at 3m should produce a rotation of 0 deg")
testutils.confirm_equal( 0, nps.mag(implied_Rt10[3,:]),
eps = 0.01,
msg = "diff(model,model) at 3m should produce a translation of 0 m")
########## Check outlier handling when computing diffs without uncertainties.
########## The model may only fit in one regions. Using data outside of that
########## region poisons the solve unless we treat those measurements as
########## outliers. This is controlled by the f_scale parameter in
########## implied_Rt10__from_unprojections().
difflen, diff, q0, implied_Rt10 = \
mrcal.projection_diff( (model_opencv8,model_splined),
gridn_width = gridn_width,
distance = 5,
use_uncertainties = False,
focus_radius = 800)
testutils.confirm_equal( 0, difflen[icenter[0],icenter[1]],
eps = 0.1,
msg = "Low-enough diff with high focus_radius")
difflen, diff, q0, implied_Rt10 = \
mrcal.projection_diff( (model_opencv8,model_splined),
gridn_width = gridn_width,
distance = 5,
use_uncertainties = False,
focus_radius = 366)
testutils.confirm_equal( 0, difflen[icenter[0],icenter[1]],
eps = 0.1,
msg = "Low-enough diff with low focus_radius")
########## Check that the solver is willing to move the origin around freely to
########## compute a very tight fit. I'm seeing that the 'trf' solver doesn't
########## like doing that and that it finds a highly suboptimal implied
########## transformation
# I generate a model with a focal length shifted anisotropically. This sounds
# weird, but is representative of the variation I see in real-life solves
model_opencv8_shiftedz = mrcal.cameramodel(model_opencv8)
lensmodel,intrinsics = model_opencv8_shiftedz.intrinsics()
intrinsics[0] *= 1.0001
intrinsics[1] *= 1.0002
model_opencv8_shiftedz.intrinsics(intrinsics = (lensmodel,intrinsics))
difflen, diff, q0, implied_Rt10 = \
mrcal.projection_diff( (model_opencv8,model_opencv8_shiftedz),
gridn_width = gridn_width,
distance = 50000,
use_uncertainties = False,
focus_radius = 1500)
testutils.confirm_equal( 0, difflen[icenter[0],icenter[1]],
eps = 2e-2,
msg = "implied_Rt10 solver moves translation sufficiently. Looking at difflen at center")
testutils.confirm_equal( 0, np.mean(difflen),
eps = .2,
msg = "implied_Rt10 solver moves translation sufficiently. Looking at mean(difflen)")
testutils.finish()
|
