pytorch-tensorboard的基本用法二——add_image()

pytorch-tensorboard的基本用法二——add_image()

1、add_image简介和注意

注意：
(1) dataformats, 需要先观察再填写，默认的为CHW，有时不一定正确
(2) 使用PIL.Image打开的图片类型是’PIL.JpegImagePlugin.JpegImageFile’
，与要求的tensor 或者numpy类型不符合
(3) global step 是记录每一步操作的标识符

2、代码实现

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

if __name__ == '__main__':
    writer = SummaryWriter("logs")
    image_path = "data/hymenoptera_data/train/bees/16838648_415acd9e3f.jpg"
    img_PIL = Image.open(image_path)
    # print(type(img_PIL)) 
    img_array = np.array(img_PIL)
    print(img_array.shape)
    print(type(img_array))
    writer.add_image("test",img_array, 2, dataformats='HWC')

    writer.close()

3、结果

tag 相同的可以通过拖动上方的按钮来观察