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

預期結果：