Class AlignmentParams

src/sharp/models/params.py:36–51

Parameters for depth alignment.

@dataclasses.dataclass
class AlignmentParams:
    """Parameters for depth alignment."""

    kernel_size: int = 16
    stride: int = 1
    frozen: bool = False

    # The following parameters are only used for LearnedAlignment.
    # Number of steps in the UNet for LearnedAlignment.
    steps: int = 4
    # Activation type for LearnedAlignment.
    activation_type: math_utils.ActivationType = "exp"
    # Whether to use depth decoder features for LearnedAlignment.
    depth_decoder_features: bool = False
    # Base width of the UNet for LearnedAlignment.
    base_width: int = 16


@dataclasses.dataclass