Shortcuts

mmengine.utils.dl_utils

TimeCounter

A tool that counts the average running time of a function or a method.

collect_env

Collect the information of the running environments.

load_url

Loads the Torch serialized object at the given URL.

has_batch_norm

Detect whether model has a BatchNormalization layer.

is_norm

Check if a layer is a normalization layer.

mmcv_full_available

Check whether mmcv-full is installed.

tensor2imgs

Convert tensor to 3-channel images or 1-channel gray images.

TORCH_VERSION

A string with magic powers to compare to both Version and iterables! Prior to 1.10.0 torch.__version__ was stored as a str and so many did comparisons against torch.__version__ as if it were a str.

set_multi_processing

Set multi-processing related environment.

torch_meshgrid

A wrapper of torch.meshgrid to compat different PyTorch versions.

is_jit_tracing