def align_reconstructions(reconstruction_shots,
reconstruction_name,
use_points_constraints,
camera_constraint_type='soft_camera_constraint'):
ra = csfm.ReconstructionAlignment()
if camera_constraint_type is 'soft_camera_constraint':
add_camera_constraints_soft(ra, reconstruction_shots,
reconstruction_name)
if camera_constraint_type is 'hard_camera_constraint':
add_camera_constraints_hard(ra, reconstruction_shots,
reconstruction_name, True)
if use_points_constraints:
add_point_constraints(ra, reconstruction_shots, reconstruction_name)
logger.info("Running alignment")
ra.run()
logger.info(ra.brief_report())
transformations = {}
for key in reconstruction_shots:
rec_name = reconstruction_name(key)
r = ra.get_reconstruction(rec_name)
s = r.scale
A = cv2.Rodrigues(np.array([r.rx, r.ry, r.rz]))[0]
b = np.array([r.tx, r.ty, r.tz])
transformations[key] = invert_similarity(s, A, b)
return transformations
def test_two_reconstructions_common_camera():
"""Two reconstructions"""
ra = csfm.ReconstructionAlignment()
ra.add_shot('a_1', 0, 0, 0, -1, 0, 0, True)
ra.add_shot('a_2', 0, 0, 0, -2, 0, 0, True)
ra.add_shot('a_3', 0, 0, 0, 0, 0, 0, True)
ra.add_shot('a_4', 0, 0, 0, 0, -1, 0, True)
ra.add_shot('a_5', 0, 0, 0, -1, 0, 0, True)
ra.add_reconstruction('a', 0, 0, 0, 0, 0, 0, 1, True)
ra.add_shot('b_3', 0, 0, 0, -1, -1, 0, True)
ra.add_shot('b_4', 0, 0, 0, -1, -2, 0, True)
ra.add_shot('b_5', 0, 0, 0, -2, -1, 0, True)
ra.add_shot('b_6', 0, 0, 0, -4, 0, 0, True)
ra.add_shot('b_7', 0, 0, 0, -5, 0, 0, True)
ra.add_reconstruction('b', 0, 0, 0, 0, 0, 0, 1, False)
ra.add_common_camera_constraint('a', 'a_3', 'b', 'b_3', 1, 1)
ra.add_common_camera_constraint('a', 'a_4', 'b', 'b_4', 1, 1)
ra.add_common_camera_constraint('a', 'a_5', 'b', 'b_5', 1, 1)
ra.run()
rec_a = ra.get_reconstruction('a')
rec_b = ra.get_reconstruction('b')
assert np.allclose(get_reconstruction_origin(rec_a), [0, 0, 0], atol=1e-6)
assert np.allclose(get_reconstruction_origin(rec_b), [-1, -1, 0],
atol=1e-6)
assert np.allclose(rec_a.scale, 1)
assert np.allclose(rec_b.scale, 1)
def test_two_reconstructions_rigid_alignment():
"""Two reconstructions"""
ra = csfm.ReconstructionAlignment()
ra.add_shot('a_1', 0, 0, 0, -1, 0, 0, True)
ra.add_shot('a_2', 0, 0, 0, -2, 0, 0, True)
ra.add_shot('a_3', 0, 0, 0, 0, 0, 0, True)
ra.add_shot('a_4', 0, 0, 0, 0, -1, 0, True)
ra.add_shot('a_5', 0, 0, 0, -1, 0, 0, True)
ra.add_reconstruction('a', 0, 0, 0, 0, 0, 0, 1, False)
ra.add_shot('b_3', 0, 0, 0, -1, -1, 0, True)
ra.add_shot('b_4', 0, 0, 0, -1, -2, 0, True)
ra.add_shot('b_5', 0, 0, 0, -2, -1, 0, True)
ra.add_shot('b_6', 0, 0, 0, -4, 0, 0, True)
ra.add_shot('b_7', 0, 0, 0, -5, 0, 0, True)
ra.add_reconstruction('b', 0, 0, 0, 0, 0, 0, 1, False)
ra.add_relative_absolute_position_constraint('a', 'a_3', 5, 5, 0, 1)
ra.add_relative_absolute_position_constraint('a', 'a_4', 5, 6, 0, 1)
ra.add_relative_absolute_position_constraint('b', 'b_3', 5, 5, 0, 1)
ra.add_relative_absolute_position_constraint('b', 'b_4', 5, 6, 0, 1)
ra.run()
rec_a = ra.get_reconstruction('a')
rec_b = ra.get_reconstruction('b')
assert np.allclose(get_reconstruction_origin(rec_a), [5, 5, 0], atol=1e-6)
assert np.allclose(get_reconstruction_origin(rec_b), [4, 4, 0], atol=1e-6)
assert np.allclose(rec_a.scale, 1)
assert np.allclose(rec_b.scale, 1)
def test_single_shot():
"""Single shot test."""
ra = csfm.ReconstructionAlignment()
ra.add_shot('1', 0.5, 0, 0, 0, 0, 0, False)
ra.add_absolute_position_constraint('1', 1, 0, 0, 1)
ra.run()
s1 = ra.get_shot('1')
assert np.allclose(get_shot_origin(s1), [1, 0, 0], atol=1e-6)
def test_singleton_reconstruction():
"""Single shot in a single reconstruction."""
ra = csfm.ReconstructionAlignment()
ra.add_shot('1', 0, 0, 0, 0, 0, 0, False)
ra.add_reconstruction('a', 0, 0, 0, 0, 0, 0, 4, False)
ra.add_relative_motion_constraint(
csfm.RARelativeMotionConstraint('a', '1', 0, 0, 0, -1, 0, 0))
ra.add_absolute_position_constraint('1', 1, 0, 0, 1)
ra.run()
s1 = ra.get_shot('1')
assert np.allclose(get_shot_origin(s1), [1, 0, 0], atol=1e-6)
def test_common_points():
"""Two reconstructions, two common points"""
ra = csfm.ReconstructionAlignment()
ra.add_reconstruction('a', 0, 0, 0, 0, 0, 0, 1, True)
ra.add_reconstruction('b', 0, 0, 0, 0, 0, 0, 1, False)
ra.add_common_point_constraint('a', 0, 0, 0, 'b', -1, 0, 0, 1.0)
ra.add_common_point_constraint('a', 1, 0, 0, 'b', 0, 0, 0, 1.0)
ra.run()
rec_b = ra.get_reconstruction('b')
o_b = get_reconstruction_origin(rec_b)
assert np.allclose(o_b, [1, 0, 0], atol=1e-6)
def align_reconstructions(reconstruction_shots):
ra = csfm.ReconstructionAlignment()
add_camera_constraints(ra, reconstruction_shots)
add_point_constraints(ra, reconstruction_shots)
ra.run()
transformations = {}
for key in reconstruction_shots:
rec_name = encode_reconstruction_name(key)
r = ra.get_reconstruction(rec_name)
s = r.scale
A = cv2.Rodrigues(np.array([r.rx, r.ry, r.rz]))[0]
b = np.array([r.tx, r.ty, r.tz])
transformations[key] = invert_similarity(s, A, b)
return transformations
def test_two_reconstructions():
"""Two reconstructions"""
ra = csfm.ReconstructionAlignment()
ra.add_shot('1', 0, 0, 0, 0, 0, 0, False)
ra.add_shot('2', 0, 0, 0, 0, 0, 0, False)
ra.add_shot('3', 0, 0, 0, 0, 0, 0, False)
ra.add_shot('4', 0, 0, 0, 0, 0, 0, False)
ra.add_reconstruction('a', 0, 0, 0, 0, 0, 0, 1, False)
ra.add_relative_motion_constraint(
csfm.RARelativeMotionConstraint('a', '1', 0, 0, 0, 0, 0, 0))
ra.add_relative_motion_constraint(
csfm.RARelativeMotionConstraint('a', '2', 0, 0, 0, -1, 0, 0))
ra.add_relative_motion_constraint(
csfm.RARelativeMotionConstraint('a', '3', 0, 0, 0, -2, 0, 0))
ra.add_reconstruction('b', 0, 0, 0, 0, 0, 0, 1, False)
ra.add_relative_motion_constraint(
csfm.RARelativeMotionConstraint('b', '2', 0, 0, 0, 0, 0, 0))
ra.add_relative_motion_constraint(
csfm.RARelativeMotionConstraint('b', '3', 0, 0, 0, -1, 0, 0))
ra.add_relative_motion_constraint(
csfm.RARelativeMotionConstraint('b', '4', 0, 0, 0, -2, 0, 0))
ra.add_absolute_position_constraint('1', 1, 0, 0, 1)
ra.add_absolute_position_constraint('2', 2, 0, 0, 1)
ra.run()
s1 = ra.get_shot('1')
s2 = ra.get_shot('2')
s3 = ra.get_shot('3')
s4 = ra.get_shot('4')
rec_a = ra.get_reconstruction('a')
rec_b = ra.get_reconstruction('b')
assert np.allclose(get_shot_origin(s1), [1, 0, 0], atol=1e-6)
assert np.allclose(get_shot_origin(s2), [2, 0, 0], atol=1e-6)
assert np.allclose(get_shot_origin(s3), [3, 0, 0], atol=1e-6)
assert np.allclose(get_shot_origin(s4), [4, 0, 0], atol=1e-6)
assert np.allclose(get_reconstruction_origin(rec_a), [1, 0, 0], atol=1e-6)
assert np.allclose(get_reconstruction_origin(rec_b), [2, 0, 0], atol=1e-6)
assert np.allclose(rec_a.scale, 1)
assert np.allclose(rec_b.scale, 1)
def test_pair():
"""Simple single reconstruction two shots test."""
ra = csfm.ReconstructionAlignment()
ra.add_shot('1', 0, 0, 0, 0, 0, 0, False)
ra.add_shot('2', 0, 0, 0, 0, 0, 0, False)
ra.add_reconstruction('a', 0, 0, 0, 0, 0, 0, 4, False)
ra.add_relative_motion_constraint(
csfm.RARelativeMotionConstraint('a', '1', 0, 0, 0, 0, 0, 0))
ra.add_relative_motion_constraint(
csfm.RARelativeMotionConstraint('a', '2', 0, 0, 0, -1, 0, 0))
ra.add_absolute_position_constraint('1', 1, 0, 0, 1)
ra.add_absolute_position_constraint('2', 3, 0, 0, 1)
ra.run()
s1 = ra.get_shot('1')
s2 = ra.get_shot('2')
rec_a = ra.get_reconstruction('a')
assert np.allclose(get_shot_origin(s1), [1, 0, 0], atol=1e-6)
assert np.allclose(get_shot_origin(s2), [3, 0, 0], atol=1e-6)
assert np.allclose(get_reconstruction_origin(rec_a), [1, 0, 0], atol=1e-6)
assert np.allclose(rec_a.scale, 0.5)
def test_two_shots_one_fixed():
"""Two shot, one reconstruction. One shot is fixed"""
ra = csfm.ReconstructionAlignment()
ra.add_shot('1', 0, 0, 0, -1, 0, 0, True)
ra.add_shot('2', 0, 0, 0, 0, 0, 0, False)
ra.add_reconstruction('a', 0, 0, 0, 0, 0, 0, 1, False)
ra.add_relative_motion_constraint(
csfm.RARelativeMotionConstraint('a', '1', 0, 0, 0, 0, 0, 0))
ra.add_relative_motion_constraint(
csfm.RARelativeMotionConstraint('a', '2', 0, 0, 0, -1, 0, 0))
# Next line should be ignored because shot 1 is fixed
ra.add_absolute_position_constraint('1', 100, 0, 0, 1)
ra.add_absolute_position_constraint('2', 3, 0, 0, 1)
ra.run()
s1 = ra.get_shot('1')
s2 = ra.get_shot('2')
rec_a = ra.get_reconstruction('a')
assert np.allclose(get_shot_origin(s1), [1, 0, 0], atol=1e-6)
assert np.allclose(get_shot_origin(s2), [3, 0, 0], atol=1e-6)
assert np.allclose(get_reconstruction_origin(rec_a), [1, 0, 0], atol=1e-6)
assert np.allclose(rec_a.scale, 0.5)
