代码实战（小土堆）——TensorBoard的使用（二）

代码

from torch.utils.tensorboard import SummaryWriter
import numpy as np
from PIL import Image

writer = SummaryWriter("logs")

# 获取图片地址
img_path ="data/train/ants_image/0013035.jpg"
# 读取图片（类型是PIL，不满足要求）
img_PIL = Image.open(img_path)
# 使用numpy类型
img_array = np.array(img_PIL)
print(type(img_array))
print(img_array.shape)

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

writer.close()

add_image()的使用

从PIL到numpy，需要在add_image()中指定shape中每一个数字/维表示的含义。

• 绘制图片，可用于检查模型的输入，监测 feature map 的变化，或是观察 weight。
• tag：名称
• img_tensor：类型
• global_step：第几张
• dataformats：tensor是CHW，numpy是HWC。