Pytorch学习——Tensorboard

Tensorboard是可视化工具。

1.add_scalar

from torch.utils.tensorboard import SummaryWriter

writer = SummaryWriter("logs")

# writer.add_image()
for i in range(100):
    writer.add_scalar("y=2x", i, 2*i)

writer.close()

找到生成文件所在位置：

点开网址：

2.add_image

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

writer = SummaryWriter("logs")
image_path = "data/train/ants_image/0013035.jpg"
img_PIL = Image.open(image_path)
img_array = np.array(img_PIL)  #转换类型

writer.add_image("img", img_array, 1, dataformats='HWC' )

writer.close()