Tensorboard

import cv2
import torchvision
import numpy as np
from torch.utils.tensorboard import SummaryWriter
from torch.utils.tensorboard import SummaryWriter

#开启
writer = SummaryWriter("logs")

#使用OpenCV读取图像（BGR格式）
img_bgr = cv2.imread("图片路径")
#将图像从BGR格式转换为RGB格式
img_rgb = cv2.cvtColor(img_bgr, cv2.COLOR_BGR2RGB)
#转化为Tensor
img = torchvision.transforms.ToTensor()(img_rgb)
#添加图片
writer.add_image("Picture", img)

#添加函数
for i in range(100):
    writer.add_scalar("y=2x", 2 * i, i)

#关闭
writer.close()

查看

tensorboard --logdir D:\study\anaconda\JupyterData\pytorch\logs