"""
本节主要学会使用tensorboard如何可视化照片
"""
from torch.utils.tensorboard import SummaryWriter
import numpy as np
from PIL import Image
writer = SummaryWriter('add_image_logs')
image_path = 'dataset/train/ants/0013035.jpg'
img_PIL = Image.open(image_path)
img_array = np.array(img_PIL)
writer.add_image('test',img_array,1,dataformats='HWC') #从PIL到numpy的转换需要指定三个值的HWC的位置
for i in range(100):
writer.add_scalar('y=2x',3*i,i) #title/y/x
writer.close()
pytorch入门3:学会使用tensorboard如何可视化照片
最新推荐文章于 2024-07-11 22:27:50 发布