Pytorch 学习TensorBoard的使用

def add_scalar 函数

绘制y=x 轴曲线

from torch.utils.tensorboard import SummaryWriterwriter =SummaryWriter("logs")for i in range(100):writer.add_scalar("y=x",i,i)
writer.close()

使用 tensorboard --logdir=logs --port=6007 来读取logs 的内容

def add_image 函数

读取图片

from torch.utils.tensorboard import SummaryWriter
from PIL import Image
import numpy as npwriter =SummaryWriter("logs")
image_path="data/train/ants_image/0013035.jpg"
Img_PIL =Image.open(image_path)
#img.show()
print(type(Img_PIL))
# 转换类型为numpy
img_arry=np.array(Img_PIL)
print(type(img_arry))
print(img_arry.shape)writer.add_image("test",img_arry,1,dataformats='HWC')writer.close()