Tensorboard的使用（二）：add_image()的使用

from torch.utils.tensorboard import SummaryWriter
import numpy as np
from PIL import Image
writer = SummaryWriter("../../../logs")

# 图片路径
imagr_path = "D:\\PycharmProjects\\learn_torch\\dataset\\train\\ants\\5650366_e22b7e1065.jpg"

# 打开图片
img = Image.open(imagr_path)

# 格式转换
img_array = np.array(img)

writer.add_image("test", img_array, 1, dataformats='HWC')
writer.close()

add_image()添加的图片的格式：所以要进行一下转换 img_array = np.array(img)

【注意】：从PIL到numpy，需要在add_image()中指定shape中每一个数字的含义

dataformats='HWC'

运行指令：tensorboard --logdir=logs

 

 【补充】：1.利用Opencv读取图片，获得numpy类型图片数据

                    2.利用numpy.array()，对PIL图片进行转换