Class InitializerParams

src/sharp/models/params.py:70–102

Parameters for initializer.

@dataclasses.dataclass
class InitializerParams:
    """Parameters for initializer."""

    # Common parameters.
    # Multiply scale of Gaussians by this factor.
    scale_factor: float = 1.0
    # Factor to convert inverse depth to disparity.
    disparity_factor: float = 1.0
    # Stride of the initializer.
    stride: int = 2

    # Parameters that only affect MultiLayerInitializer.
    # How many layers of Gaussians to predict (only available for MultiLayerInitializer).
    num_layers: int = 2
    # Which option to use for depth initialization.
    first_layer_depth_option: DepthInitOption = "surface_min"
    rest_layer_depth_option: DepthInitOption = "surface_min"
    # Which option to use for color initialization.
    color_option: ColorInitOption = "all_layers"
    # Which depth value to use for depth layers.
    base_depth: float = 10.0
    # Deactivate gradient for feature inputs.
    feature_input_stop_grad: bool = False
    # Whether to normalize depth to [DepthTransformParam.depth_min,
    # DepthTransformParam.depth_max).
    normalize_depth: bool = True

    # Output only the inpainted layer. In this case, num_layers = 1.
    output_inpainted_layer_only: bool = False
    # Whether to set the uninpainted region to zero opacities.
    set_uninpainted_opacity_to_zero: bool = False
    # Whether to concatenate the inpainting mask to the feature input.
    concat_inpainting_mask: bool = False


@dataclasses.dataclass