Tensorboard的使用

# tensorboard的安装：pip install tensorboard
# 修改tensorboard端口号：tensorboard --logdir=logs --port=6007
# 查看计算图：tensorboard --logdir=logs

# TensorBoard的使用
from torch.utils.tensorboard import SummaryWriter
import numpy as np
from PIL import Image

writer = SummaryWriter("logs")  # 创建实例  指定目录
image_path="data/train/ants_image/28847243_e79fe052cd.jpg"
img_PIL=Image.open(image_path)  # 打开图片
img_array=np.array(img_PIL)  # 将img转成numpy类型
print(type(img_array))
print(img_array.shape)

# writer.add_image("test",img_array,1,dataformats='HWC')
writer.add_image("train",img_array,1,dataformats='HWC')

# y=2x
for i in range(100):
    # 第一个参数是tag标题  第二个参数是y轴的值 第三个参数是x轴的值
    writer.add_scalar("y=2x",2*i,i)  # 添加标量，数
writer.close()