写入函数:writer.add_scalar()
from torch.utils.tensorboard import SummaryWriterwriter = SummaryWriter('logs')for i in range(100):writer.add_scalar('y=2x',2*i,i)writer.close()
写入图片:writer.add_image()
from torch.utils.tensorboard import SummaryWriterfrom PIL import Imageimport numpy as npwriter = SummaryWriter('logs')img_path = 'data/dataset/train/bees/16838648_415acd9e3f.jpg' #读取图片路径img_PIL = Image.open(img_path) #用PIL格式打开img_array =np.array(img_PIL) #用np转换成数组格式print(type(img_array)) #查看转换后的格式类型print(img_array.shape) #查看图片尺寸和通道格式writer.add_image('train',img_array,2,dataformats='HWC')# img_tensor: Default is :math:`(3, H, W)# 上一步print(img_array.shape)查看图片为:(H, W, 3)# 官方给出的help中查看给出的example,加上dataformats='HWC'# for i in range(100):# writer.add_scalar('y=2x',2*i,i)writer.close()
命令行窗口键入 tensorboard —logdir=logs即可查看
