Tensorboard学习（2.绘制image图像）

利用Tensorboard，绘制image图像进行展示

1.代码参考：

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

# 记录图像函数add_image。
writer = SummaryWriter("test_image")

img_path = "D:\深度学习资料\\032Pytorch\PyTorch 教程-小土堆\数据集\hymenoptera_data\hymenoptera_data\\train\\ants\\0013035.jpg"
img = Image.open(img_path)

# writer数据格式要求，需要对图像进行格式转换
img_np = np.array(img)

writer.add_image("image", img_np, 1, dataformats="HWC")

writer.close()

2. writer.add_image的格式要求

① 图像数据格式（opencv、numpy等）

② 图像维度（Height*wide*Channel需要根据需要调整）

3.代码运行后生成文件夹“test_image”,如下：

4. 在cmd命令行中进行展示，命令为：

tensorboard --logdir=test_image

结果展示为：

点击链接：

注：有的时候在绘图的时候会出现，以下bug：

原因为：PIL版本过高，不合适。重新安装低版本的Pillow即可。如：

再次运行，即可成功运行，并生成图像。