Extract features from images.

Extract features from images.

Estimate disparity between pair of frames

Forward function for `FFN`. The function would add x to the output tensor if residue is None.

Placeholder of forward function.

Placeholder of forward function.

Forward for head part which is called after every decoder layer. Args: decoder_out (Tensor): in shape (num_queries, batch_size, c

Args: imgs (List[Tensor]): the outer list indicates test-time augmentations and inner Tensor should have a shape

Args: imgs (List[Tensor]): the outer list indicates test-time augmentations and inner Tensor should have a shape

Forward function for training. Args: img (Tensor): Input images. img_metas (list[dict]): List of image info dict wher

input_resolution (tuple[int]): The height and width of input window_size (tuple[int]): The height, width and depth of window

Description: Generates "sample_count" number of disparity samples from the sea

Description: Generates the disparity search range. Returns: :min_disparity: Lower bound of disparity search range

this will perform the index select, cat the tensors, and provide the attn_bias from cache

this will perform the index select, cat the tensors, and provide the attn_bias from cache

Returns: The rank of the current process within the global process group.

Returns: The number of processes in the process group

Compute classification and mask targets for all images for a decoder layer. Args: cls_scores_list (list[Tensor]): Mask sc

Inverse function of sigmoid. Args: x (Tensor): The tensor to do the inverse. eps (float): EPS avoid numerical

Just like `yaml.load(open(filename))`, but inherit attributes from its `__BASE__`. Args: filename (str or file-like obje

Apply function to a list of arguments. Note: This function applies the ``func`` to multiple inputs and map the multiple outputs o

@img: (B,C,H,W) in range 0-255, RGB order

Sample positive and negative bboxes. This is a simple implementation of bbox sampling given candidates, assigning results and ground

Generate grid Points of a single level. Note: This function is usually called by method ``self.grid_priors``. Args:

Upsample input tensor by a factor of 2

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