-
SummaryWriter()将什么写入文件中,如果不指定的话,就写入默认的
- 需要两个方法
writer.add_image() wruter.add_scalar()
-
from torch.utils.tensorboard import SummaryWriter # 导入tensorboard,再导入SummaryWriter这个类 writer = SummaryWriter("logs") # 利用OpenCV读取图片,获得numpy型图片数据 ''' ima_path = "路径名" from PIL import Image img = Image.open(image_path) print(type(img)) (输出的是PIL型的图片,要转换成tensor的或者numpy的) 利用numpy。array(),对PIL图片进行转换: 首先导入numpy import numpy as np img_array = np.array(img) print(type(img_array)) (可能会报错) ''' # 在这行代码中加入参数 (dataformats=’HWC‘) writer.add_image() # y = x for i in range(100): writer.add_scalar("y = 2x", 2*i, i) writer.close()
- 需要两个方法