mask from (N, L) into (N, 1(H), 1(L), L) and make it negative

Forward function for plugins.

Check whether the object has a method. Args: method (str): The method name to check. obj (object): The object to check. Retu

A property to get the DDP-wrapped model on the device.

A wrapper around :func:`grid_sample` to support 3D point_coords tensors Unlike :func:`torch.nn.functional.grid_sample` it assumes point_coords to

Calculate annealing cos learning rate. Cosine anneal from `weight * start + (1 - weight) * end` to `end` as percentage goes from 0.0 to 1.0.

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

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

Read data from a given ``filepath`` with 'rb' mode. Note: There are two types of return values for ``get``, one is ``bytes``

Check whether it is a sequence of some type. Args: seq (Sequence): The sequence to be checked. expected_type (type): Expected typ

Check whether it is a tuple of some type. A partial method of :func:`is_seq_of`.

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

r"""Register a path to registry with key 'name' Args: name: Key with which the path will be registered. Usage:

Train for a given number of iterations

Apply element-wise weight and reduce loss. Args: loss (Tensor): Element-wise loss. weight (Tensor): Element-wise weights.

Initialize BaseModule, inherited from `torch.nn.Module`

Init. Args: scale_factor (float): scaling mode (str): interpolation mode

Convert the type and range of the input image. It converts the input image to np.float32 type and range of [0, 1]. It is mainly used for pre-

Convert the type and range of the image according to dst_type. It converts the image to desired type and range. If `dst_type` is np.uint8, im

Return an activation function given a string

Average latest n values or all values.

Build projection layer for key/query/value/out.

Crop from ``img``

Calculate evaluation metrics Args: results (list[ndarray] | list[str]): List of prediction segmentation maps or list of predic

Evaluate the results. Args: runner (:obj:`mmcv.Runner`): The underlined training runner. results (list): Output resul

Get the registry record. Args: key (str): The class name in string format. Returns: class: The corresponding

Initialize and get a logger by name. If the logger has not been initialized, this method will initialize the logger by adding one or two hand

Read an image. Args: img_or_path (ndarray or str or Path): Either a numpy array or str or pathlib.Path. If it is a numpy arra

Load checkpoint as specified in the config file. If load_finetuned is True, load the finetuned model; otherwise, load the pretrained

Make divisible function. This function rounds the channel number to the nearest value that can be divisible by the divisor. It is taken from

Print a list of dictionaries as a table to a string.

Save ndarray to local numpy file. Args: array (ndarray): Ndarray to save. temp_file_name (str): Numpy file name. If 'temp_file_na