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

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

用法:

add_images(tag, img_tensor, global_step=None, walltime=None, dataformats='NCHW')

参数：

• tag(string) -数据标识符

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

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

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

• dataformats(string) -NCHW、NHWC、CHW、HWC、HW、WH等形式的图像数据格式规范。

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

请注意，这需要pillow 包。

形状：

img_tensor：默认为 (N, 3, H, W) 。如果指定dataformats，则接受其他形状。例如NCHW 或 NHWC。

例子：

from torch.utils.tensorboard import SummaryWriter
import numpy as np

img_batch = np.zeros((16, 3, 100, 100))
for i in range(16):
    img_batch[i, 0] = np.arange(0, 10000).reshape(100, 100) / 10000 / 16 * i
    img_batch[i, 1] = (1 - np.arange(0, 10000).reshape(100, 100) / 10000) / 16 * i

writer = SummaryWriter()
writer.add_images('my_image_batch', img_batch, 0)
writer.close()

预期结果：