Function create_monodepth_decoder

src/sharp/models/decoders/monodepth_decoder.py:18–37

Create DepthDensePredictionTransformer model. Args: patch_encoder_preset: The preset patch encoder architecture in SPN. dims_decoder: The decoder architecture.

(
    patch_encoder_preset: ViTPreset,
    dims_decoder=None,
)

def create_monodepth_decoder(
    patch_encoder_preset: ViTPreset,
    dims_decoder=None,
) -> MultiresConvDecoder:
    """Create DepthDensePredictionTransformer model.

    Args:
        patch_encoder_preset: The preset patch encoder architecture in SPN.
        dims_decoder: The decoder architecture.
    """
    dims_encoder = MONODEPTH_ENCODER_DIMS_MAP[patch_encoder_preset]
    if dims_decoder is None:
        dims_decoder = dims_encoder[0]
    if isinstance(dims_decoder, int):
        dims_decoder = [dims_decoder]
    decoder = MultiresConvDecoder(
        dims_encoder=[dims_decoder[0]] + list(dims_encoder), dims_decoder=dims_decoder
    )

    return decoder