Shortcuts

mmengine.utils.apply_to

mmengine.utils.apply_to(data, expr, apply_func)[source]

Apply function to each element in dict, list or tuple that matches with the expression.

For examples, if you want to convert each element in a list of dict from np.ndarray to Tensor. You can use the following code:

Examples

>>> from mmengine.utils import apply_to
>>> import numpy as np
>>> import torch
>>> data = dict(array=[np.array(1)]) # {'array': [array(1)]}
>>> result = apply_to(data, lambda x: isinstance(x, np.ndarray), lambda x: torch.from_numpy(x))
>>> print(result) # {'array': [tensor(1)]}
Parameters
  • data (Any) – Data to be applied.

  • expr (Callable) – Expression to tell which data should be applied with the function. It should return a boolean.

  • apply_func (Callable) – Function applied to data.

Returns

The data after applying.

Return type

Any

Read the Docs v: v0.8.3
Versions
latest
stable
v0.8.3
v0.8.2
v0.8.1
v0.8.0
v0.7.4
v0.7.3
v0.7.2
v0.7.1
v0.7.0
v0.6.0
v0.5.0
v0.4.0
v0.3.0
v0.2.0
Downloads
On Read the Docs
Project Home
Builds

Free document hosting provided by Read the Docs.