AngleSelect

class imgreg.models.angleselect.solver.AngleSelect(img: numpy.ndarray, ref_img: numpy.ndarray, angle_a: float, angle_b: float, upsampling: int = 10)[source]

Implements a model for angle selection

Parameters
imgnumpy.ndarray

The input image (one color channel only).

ref_imgnumpy.ndarray

The reference image (one color channel only).

angle_a, angle_bfloat

Candidate angles A and B in degrees

upsamplingint

Upsampling factor.

1 => no upsampling, 20 => precision to 1/20 of a pixel.

Notes

The model implements the following dependency graph to construct it’s Parameters.

// dependencies digraph { UPSAMPLING [shape=oval] SELECTOR [shape=oval] ROT_A_IMG [shape=box] ROT_TR_B_IMG [shape=box] ROT_TR_A_IMG [shape=box] TRANSLATION_B [shape=oval] REF_IMG [shape=box] IMG [shape=box] ANGLE_A [shape=oval] ANGLE_B [shape=oval] RECOVERED_ROT_TR_IMG [shape=box] RECOVERED_TRANSLATION [shape=oval] TRANSLATION_A [shape=oval] RECOVERED_ROT_IMG [shape=box] ROT_B_IMG [shape=box] RECOVERED_ROTATION [shape=oval] REF_IMG -> SELECTOR ROT_TR_A_IMG -> SELECTOR ROT_TR_B_IMG -> SELECTOR ANGLE_A -> ROT_A_IMG IMG -> ROT_A_IMG ANGLE_B -> ROT_TR_B_IMG IMG -> ROT_TR_B_IMG TRANSLATION_B -> ROT_TR_B_IMG ANGLE_A -> ROT_TR_A_IMG TRANSLATION_A -> ROT_TR_A_IMG IMG -> ROT_TR_A_IMG REF_IMG -> TRANSLATION_B ANGLE_B -> TRANSLATION_B ROT_B_IMG -> TRANSLATION_B UPSAMPLING -> TRANSLATION_B ROT_TR_A_IMG -> RECOVERED_ROT_TR_IMG ROT_TR_B_IMG -> RECOVERED_ROT_TR_IMG SELECTOR -> RECOVERED_ROT_TR_IMG TRANSLATION_A -> RECOVERED_TRANSLATION SELECTOR -> RECOVERED_TRANSLATION TRANSLATION_B -> RECOVERED_TRANSLATION REF_IMG -> TRANSLATION_A ANGLE_A -> TRANSLATION_A ROT_A_IMG -> TRANSLATION_A UPSAMPLING -> TRANSLATION_A ROT_A_IMG -> RECOVERED_ROT_IMG SELECTOR -> RECOVERED_ROT_IMG ROT_B_IMG -> RECOVERED_ROT_IMG ANGLE_B -> ROT_B_IMG IMG -> ROT_B_IMG ANGLE_B -> RECOVERED_ROTATION ANGLE_A -> RECOVERED_ROTATION SELECTOR -> RECOVERED_ROTATION }

Methods

__init__(img, ref_img, angle_a, angle_b[, …])

Initialize self.

display(param_list[, title])

Fancy plot functionality for registered ImageParameters.

dot_graph([node_args_func])

Return a dot graph representation of the solver model.