0
点赞
收藏
分享

微信扫一扫

TensorBoard

艾晓雪 2022-03-15 阅读 47

TensorBoard的含义

显示训练时的曲线图与每一步的图像

TensorBoard的使用

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

image_path = "data/train/ants_image/0013035.jpg"
img_PIL = Image.open(image_path)
img_array = np.array(img_PIL)#将PIL转化为np.array
writer = SummaryWriter("logs")#存放信息的路径

writer.add_image("test", img_array, 1, dataformats='HWC')#显示图片
for i in range(100):

    writer.add_scalar("y=2x", i, i)#显示y=2x的曲线图

writer.close()
## 参数解析
	writer.add_scalar(“图表标题”,数值(y),步数(x))
	writer.add_image(“图表标题”,数值(y),步数(x),dataformats='HWC')
		将图片转化为(高度(H),宽度(W),尺寸(C))的格式
## 图表显示

	命令行输入:tensorboard --logdir=logs
	指定端口:tensorboard --logdir=logs --port=端口号
举报

相关推荐

0 条评论