Average latest n values or all values.

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

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

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

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

Scatter with support for kwargs dictionary.

Convert objects of various python types to :obj:`torch.Tensor`. Supported types are: :class:`numpy.ndarray`, :class:`torch.Tensor`, :class:`S

r"""Fills the input Tensor with values drawn from a truncated normal distribution. The values are effectively drawn from the normal distributi

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

Make each stage.

Make transition layer.

Scale the data in the corresponding channel.

Calculate accuracy according to the prediction and target. Args: pred (torch.Tensor): The model prediction, shape (N, num_class, ...)

Allreduce parameters. Args: params (list[torch.Parameters]): List of parameters or buffers of a model. coalesce (bool

Convert various input to color tuples. Args: color (:obj:`Color`/str/tuple/int/ndarray): Color inputs Returns: tuple[int]: A

Convert a video with ffmpeg. This provides a general api to ffmpeg, the executed command is:: `ffmpeg -y <pre_options> -i <in_file> <opt

Crop from ``img``

Encode images with backbone and decode into a semantic segmentation map of the same size as input.

Placeholder for Forward function for training.

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

Get class weight for loss function. Args: class_weight (list[float] | str | None): If class_weight is a str, take it as a fil

Read data from a given ``filepath`` with 'r' mode. Args: filepath (str or Path): Path to read data. encoding (str): T

Resize image to a given size. Args: img (ndarray): The input image. size (tuple[int]): Target size (w, h). return_scale (

Show an image. Args: img (str or ndarray): The image to be displayed. win_name (str): The window name. wait_time (int): V

Write image to file. Args: img (ndarray): Image array to be written. file_path (str): Image file path. params (None or li

Inference with slide/whole style. Args: img (Tensor): The input image of shape (N, 3, H, W). img_meta (dict): Image i

load checkpoint through HTTP or HTTPS scheme path. In distributed setting, this function only download checkpoint at local rank 0. Args:

make plugins for block. Args: in_channels (int): Input channels of plugin. plugins (list[dict]): List of plugins cfg

nn.Module: normalization layer after the third convolution layer

Since built-in ``any`` works only when the element of iterable is not iterable, implement the function.

perform evaluation and save ckpt.

Freeze stages param and norm stats.

Sample from fine grained features. Args: prev_output (list[Tensor]): Prediction of previous decode head. points (Tens

Sample from fine grained features. Args: x (list[Tensor]): Feature pyramid from by neck or backbone. points (Tensor):

Initialize rule, key_indicator, comparison_func, and best score. Here is the rule to determine which rule is used for key indicator w

Override default load function. AWS overrides the function _load_from_state_dict to recover weight_gamma and weight_beta if they are

merge dict ``a`` into dict ``b`` (non-inplace). Values in ``a`` will overwrite ``b``. ``b`` is copied first to avoid in-place modific

Convert a pillow image to numpy array. Args: img (:obj:`PIL.Image.Image`): The image loaded using PIL flag (str): Flags specifyin

Save the current checkpoint and delete unwanted checkpoint.

Save the best checkpoint. It will compare the score according to the compare function, write related information (best score, best ch

Rescale a size by a ratio. Args: size (tuple[int]): (w, h). scale (float | tuple(float)): Scaling factor. Returns: t

Judge whether to perform evaluation. Here is the rule to judge whether to perform evaluation: 1. It will not perform evaluation durin

Swap the parameter of model with parameter in ema_buffer.

Creates a watermark on the saved inference image. We request that you do not remove this to properly assign credit to Shi-Lab's work.