写入函数:writer.add_scalar()

    1. from torch.utils.tensorboard import SummaryWriter
    2. writer = SummaryWriter('logs')
    3. for i in range(100):
    4. writer.add_scalar('y=2x',2*i,i)
    5. writer.close()

    写入图片:writer.add_image()

    1. from torch.utils.tensorboard import SummaryWriter
    2. from PIL import Image
    3. import numpy as np
    4. writer = SummaryWriter('logs')
    5. img_path = 'data/dataset/train/bees/16838648_415acd9e3f.jpg' #读取图片路径
    6. img_PIL = Image.open(img_path) #用PIL格式打开
    7. img_array =np.array(img_PIL) #用np转换成数组格式
    8. print(type(img_array)) #查看转换后的格式类型
    9. print(img_array.shape) #查看图片尺寸和通道格式
    10. writer.add_image('train',img_array,2,dataformats='HWC')
    11. # img_tensor: Default is :math:`(3, H, W)
    12. # 上一步print(img_array.shape)查看图片为:(H, W, 3)
    13. # 官方给出的help中查看给出的example,加上dataformats='HWC'
    14. # for i in range(100):
    15. # writer.add_scalar('y=2x',2*i,i)
    16. writer.close()

    命令行窗口键入 tensorboard —logdir=logs即可查看