mmengine.utils.dl_utils.tensor2imgs

mmengine.utils.dl_utils.tensor2imgs(tensor, mean=None, std=None, to_bgr=True)

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

Parameters

• tensor (torch.Tensor) – Tensor that contains multiple images, shape ( N, C, H, W). \(C\) can be either 3 or 1. If C is 3, the format should be RGB.

• mean (tuple[float], optional) – Mean of images. If None, (0, 0, 0) will be used for tensor with 3-channel, while (0, ) for tensor with 1-channel. Defaults to None.

• std (tuple[float], optional) – Standard deviation of images. If None, (1, 1, 1) will be used for tensor with 3-channel, while (1, ) for tensor with 1-channel. Defaults to None.

• to_bgr (bool) – For the tensor with 3 channel, convert its format to BGR. For the tensor with 1 channel, it must be False. Defaults to True.

Returns

A list that contains multiple images.

Return type

list[np.ndarray]