mmengine.utils.apply_to

mmengine.utils.apply_to(data, expr, apply_func)

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:

示例

>>> 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)]}

参数:

• 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.

返回:

The data after applying.

返回类型:

Any