pytorch框架--显示方面--标签--图像 对应图

标签–图像 对应图示例

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

writer = SummaryWriter("logs")
image_path = "./data/train/Cat/21.jpg"
img_PIL = Image.open(image_path)
img_array = np.array(img_PIL)
# 虽然转为array类型，但是通道排序是不同的(RGB)，所以需要重新排序()
# 在torch内提供直接排序通道的参数 dataformats='HWC'

# 若不想拖动查看，可使用不同命名
# 参数：命名、图片、第几个 、注意转换通道，目前为单张图像，多张图像需.add_images 多加个s
writer.add_image("test", img_array, 3, dataformats='HWC')
# for i in range(100):
#     # 绘制x轴、y轴
#     writer.add_scalar("y=6x", 5*i, 2*i)

writer.close()

# 名字需要对应
# tensorboard --logdir=logs --port=6007