Flip image horizontally Parameters ---------- image : torch.Tensor [B,3,H,W] Image to be flipped Returns -------

Post-process an inverse and flipped inverse depth map Parameters ---------- inv_depth : torch.Tensor [B,1,H,W] Inverse depth

Convert the model into training mode while keep layers freezed.

Transform inputs for decoder. Args: inputs (list[Tensor]): List of multi-level img features. Returns: Tensor

Placeholder of forward function.

Args: x: (B, H, W, C) window_size (int): window size Returns: windows: (num_windows*B, window_size, window_size, C)

Args: x: (B, H, W, C) window_size (int): window size Returns: windows: (num_windows*B, window_size, window_size, C)

Check and initialize input transforms. The in_channels, in_index and input_transform must match. Specifically, when input_transform i

Load checkpoint from somewhere (modelzoo, file, url). Args: filename (str): Accept local filepath, URL, ``torchvision://xxx``,

Fuse inverse depth and flipped inverse depth maps Parameters ---------- inv_depth : torch.Tensor [B,1,H,W] Inverse depth map

Initialize the weights in backbone and heads. Args: pretrained (str, optional): Path to pre-trained weights. Defa

Load checkpoint from a file or URI. Args: model (Module): Module to load checkpoint. filename (str): Accept local filepath, URL,

Load state_dict to a module. This method is modified from :meth:`torch.nn.Module.load_state_dict`. Default value for ``strict`` is set to ``F

In distributed setting, this function only download checkpoint at local rank 0.

Forward function of PSP module.

Test function.

Upsample input tensor by a factor of 2

Args: windows: (num_windows*B, window_size, window_size, C) window_size (int): Window size H (int): Height of image

Args: windows: (num_windows*B, window_size, window_size, C) window_size (int): Window size H (int): Height of image