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()

預期結果：