3.TensorBoard

1.PyCharm输入

from torch.utils.tensorboard import SummaryWriter

writer = SummaryWriter("logs")

for i in range(50):
    writer.add_scalar("y=i^2", i*i, i)

writer.close()

2.浏览器中显示,在控制台中输入
–logdir指定目录,–port指定端口

tensorboard --logdir=logs --port=6009

先运行pycharm代码,然后运行控制台代码,最后在浏览器查看
如果图像出现乱码,删除logs的文件重新运行即可

3.add.image()

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

writer = SummaryWriter("logs")
ants_img_path = "dataset/train/ants/0013035.jpg"    # ants图片的路径
bees_img_path = "dataset/train/bees/36900412_92b81831ad.jpg"    # bees图片的路径
ants_img_PIL = Image.open(ants_img_path)    # 以Jpeg格式打开
ants_img_array = np.array(ants_img_PIL)     # 将Jpeg转为numpy
bees_img_PIL = Image.open(bees_img_path)
bees_img_array = np.array(bees_img_PIL)


# add_image参数:标题,图片路径,训练轮次,数据格式(optional)
writer.add_image("Test", ants_img_array, 1, dataformats="HWC")  # 转为np时注意设置dataformat, channel放在最后一个维度
writer.add_image("Train", bees_img_array, 1, dataformats="HWC")
for i in range(50):
    writer.add_scalar("y=i^2-7*1", i*i-7*i, i)

writer.close()