TensorBoard的使用

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

writer = SummaryWriter("logs")
#介绍常使用的两个方法：
# writer.add_image()
# writer.add_scalar()

#演示1 ：
# for i in range(100):
#     writer.add_scalar("y=x",i,i)
# writer.close()
#运行结束后终端打开 tensorboard --logdir=logs --port=6007 (端口默认6006 可选)

#演示2 ：
img_path = "dataset/train/dog/dog.jpg"
img_PIL = Image.open(img_path)
print(type(img_PIL)) #查看是否满足writer.add_image()的类型，如果使用opencv读取的直接是numpy型

img_array = np.array(img_PIL) #使用numoy转换
print(type(img_array)) #查看类型
print(img_array.shape) #查看形状
writer.add_image("test",img_array, 1, dataformats='HWC')  #dataformats='HWC'指定数据类型
writer.close()