Saves module state to `destination` dictionary. This method is modified from :meth:`torch.nn.Module._save_to_state_dict`. Args: modu

Saves module state to `destination` dictionary. This method is modified from :meth:`torch.nn.Module._save_to_state_dict`. Args: modu

Substitute variable strings to their actual values.

Convert image based absolute point coordinates to image based relative coordinates for sampling. Args: abs_img_points (Tensor): Image

Add all parameters of module to the params list. The parameters of the given module will be added to the list of param groups, with s

Called after every training epoch to evaluate the results.

Called after every training iter to evaluate the results.

Placeholder for augmentation test.

Scaling bboxes w.r.t the box center. Args: bboxes (ndarray): Shape(..., 4). scale (float): Scaling factor. clip_shape (tu

initialize conv/fc bias value according to a given probability value.

Given an input and a flow-field grid, computes the output using input values and pixel locations from grid. Supported only bilinear interpolation

Brightness distortion.

Create by split datasets inheriting torch.utils.data.Datasets. # build() can be dataset-specific. Overwrite to customize.

Builder for attention.

Builder for feed-forward network (FFN).

Build padding layer. Args: cfg (None or dict): The padding layer config, which should contain: - type (str): Layer type.

Build pixel sampler for segmentation map.

Build plugin layer. Args: cfg (None or dict): cfg should contain: type (str): identify plugin layer type. layer a

NOTE: shape arg should include spatial dim only

Build segmentor.

Builder for transformer layer.

Build upsample layer. Args: cfg (dict): The upsample layer config, which should contain: - type (str): Layer type.

Calculating square distance between a and b. Args: point_feat_a (Tensor): (B, N, C) Feature vector of each point. point_feat_b (T

CAM feature classification.

Convert camel case word into snack case. Modified from `inflection lib <https://inflection.readthedocs.io/en/latest/#inflection.under

Add check points in a single line. This method is suitable for running a task on a list of items. A timer will be registered when the method

Puts each data field into a tensor/DataContainer with outer dimension batch size. Extend default_collate to add support for :type:`~mmcv.

Collect the information of the running environments.

Collect results with CPU.

Collect results under cpu mode. On cpu mode, this function will save the results on different gpus to ``tmpdir`` and collect them by the rank

Collect results with GPU.

Collect results under gpu mode. On gpu mode, this function will encode results to gpu tensors and use gpu communication for results collectio

Compute binned relative position bias

3x3 convolution with padding.

Decay the learning rate

Get current momentums. Returns: list[float] | dict[str, list[float]]: Current momentums of all param groups. If t