Pytorch——SummaryWriter.add_image（）的使用

用法：

import cv2
from PIL import Image
from torch.utils.tensorboard import SummaryWriter
from torchvision import transforms


# 绝对路径 F:\PycharmProjects\Learning1011\dataset\train\ants\0013035.jpg
# 相对路径 dataset/train/ants/0013035.jpg

img_path = "dataset/train/ants/0013035.jpg" #   图片路径
# 读取图片的两种方式
img = Image.open(img_path)  #   方式一
# img_PIL= cv2.imread(img_path)   # 方式二


writer = SummaryWriter("logs")  #   文件夹“logs”

# 1、transforms该如何使用（pytorch）
tensor_trans = transforms.ToTensor()    #  从transforms”工具箱“中选择一个class进行创建“自己的工具”
tensor_img = tensor_trans(img)  #   根据class需要的东西传入进去，返回一个结果

writer.add_image("Tensor_img", tensor_img)
# 参数("显示图像的窗口名",显示图片的名字，1表示一张图片，两张图片为2..
writer.close()  #   关闭

显示 ：

1、在控制台输入：

tensorboard --logdir=logs#=文件名

2、 修改端口号方式：

 tensorboard --logdir=logs --port=6007#自己制定显示的端口号

结果：