pytorch学习笔记—tensorboard的使用（2）

复制图片的相对地址，打开控制台，image需要是以下三种类型

 img_tensor (torch.Tensor, numpy.array, or string/blobname)

image_path ="D:\\a\\dataset\\000.png"
from PIL import Image

是PIL.Png类型的，不满足要求。

利用opencv读取图片，获得numpy型图片数据→利用numpy.array()，对PIL图片进行转换

首先在terminal 中pip install opencv-python，安装成功后： 

 显示图片类型为numpy

from torch.utils.tensorboard import SummaryWriter
import numpy as np
from PIL import Image
writer=SummaryWriter('logs')
image_path = "D:\\a\\dataset\\000.png"
img_PIL = Image.open(image_path)
img_array = np.array(img_PIL)
print(img_array.shape)
writer.add_image("test",img_array,1,dataformats='HWC')
# y=x
for i in range(100):
    writer.add_scalar("y=x",i,i)

然后打开浏览器，得到