Python PyTorch SummaryWriter.add_image用法及代码示例

本文简要介绍python语言中 torch.utils.tensorboard.writer.SummaryWriter.add_image 的用法。

用法:

add_image(tag, img_tensor, global_step=None, walltime=None, dataformats='CHW')

参数：

• tag(string) -数据标识符

• img_tensor(torch.Tensor,numpy.array， 或者字符串/blob 名称) -图像数据

• global_step(int) -要记录的全局步长 值

• walltime(float) -事件纪元后的可选覆盖默认 walltime (time.time()) 秒

将图像数据添加到摘要中。

请注意，这需要pillow 包。

形状：

img_tensor：默认为 (3, H, W) 。您可以使用 torchvision.utils.make_grid() 将一批张量转换为 3xHxW 格式或调用 add_images 让我们完成这项工作。只要传递了相应的 dataformats 参数，带有 (1, H, W) 、 (H, W) 、 (H, W, 3) 的张量也适用，例如CHW，HWC，HW。

例子：

from torch.utils.tensorboard import SummaryWriter
import numpy as np
img = np.zeros((3, 100, 100))
img[0] = np.arange(0, 10000).reshape(100, 100) / 10000
img[1] = 1 - np.arange(0, 10000).reshape(100, 100) / 10000

img_HWC = np.zeros((100, 100, 3))
img_HWC[:, :, 0] = np.arange(0, 10000).reshape(100, 100) / 10000
img_HWC[:, :, 1] = 1 - np.arange(0, 10000).reshape(100, 100) / 10000

writer = SummaryWriter()
writer.add_image('my_image', img, 0)

# If you have non-default dimension setting, set the dataformats argument.
writer.add_image('my_image_HWC', img_HWC, 0, dataformats='HWC')
writer.close()

预期结果：