Shortcuts

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.

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

返回

A list that contains multiple images.

返回类型

list[np.ndarray]

Read the Docs v: latest
Versions
latest
stable
Downloads
pdf
html
epub
On Read the Docs
Project Home
Builds

Free document hosting provided by Read the Docs.