tensorboard
from torch.utils.tensorboard import SummaryWriter
#按住Ctrl,鼠标移到SummaryWriter上,点入即可看到官方定义
#创建实例
writer = SummaryWriter("logs")
#将事件文件存储到logs文件夹下
for i in range(100):
#使用的是add_scalar
writer.add_scalar("y=x",i,i) #标题,y轴,x轴
#writer.add_scalar("y=2x",2*i,i)
writer.close()
安装tensorboard
打开窗口
from torch.utils.tensorboard import SummaryWriter
import numpy as py
from PIL import Image
#创建实例
writer = SummaryWriter("logs")
#将事件文件存储到logs文件夹下
image_path = "图片的相对地址"
img_PIL = Image.open(image_path) #打开图片
#把图片类型转为numpy.array型
img_array = np.array(img_PIL)
print(type(img_array))
print(img_array.shape)#如果图片是(H,W,3)类型