TensorBoard是一个可视化工具,用于帮助机器学习工程师和研究人员可视化训练过程中的模型、数据和指标。
一、add_scalar:
from torch.utils.tensorboard import SummaryWriter
#文件存储到logs文件夹下
writer = SummaryWriter("logs")
#添加标量数据到Summary
#tag
#def add_scalar(self,tag,scalar_value,global_step=None,walltime=None,new_style=False,double_precision=False)
for x in range(10):
writer.add_scalar("y=2x",x,2*x)
writer.close()
二、add_image:
from PIL import Image
import numpy as np
from torch.utils.tensorboard import SummaryWriter
#文件存储到logs文件夹下
writer = SummaryWriter("logs")
image_path = "dataset/train/ants_image/0013035.jpg"
img_PIL = Image.open(image_path)
img_ndarray = np.array(img_PIL)
writer.add_image("img", img_ndarray,1,dataformats="HWC")
writer.close()