【PyTorch

【PyTorch,第1张

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

writer = SummaryWriter("logs")
image_path = "图片的路径"
img_PIL = Image.open(image_path)
img_array = np.array(img_PIL)
print(type(img_array))
print(img_array.shape)
'''
名称
必须是Numpy形式的图片
第三个参数:代表步骤
图片的形式
'''
writer.add_image("train", img_array, 1, dataformats='HWC')
# y = 2x
# 画坐标图
# for i in range(100):
    # writer.add_scalar("y=2x", 3*i, i)

writer.close()
# 在控制台输入以下命令后Ctrl+点击链接进入浏览器查看
tensorboard --logdir
# 改变端口
tensorboard --logdir= logs --port=6007

运行时必须在生成的log文件的上层目录运行,否则不能打开

No dashboards are active for the current data set.

欢迎分享,转载请注明来源:内存溢出

原文地址: http://outofmemory.cn/zaji/3973973.html

(0)
打赏 微信扫一扫 微信扫一扫 支付宝扫一扫 支付宝扫一扫
上一篇 2022-10-21
下一篇 2022-10-21

发表评论

登录后才能评论

评论列表(0条)

保存