from torch.utils.tensorboard import SummaryWriter
import numpy as np
from PIL import Image
w=SummaryWriter("logs")
img_path="dataset/train/ants/0013035.jpg"
img_PIL = Image.open(img_path)
img_array=np.array(img_PIL)
w.add_image("test",img_array,1,dataformats='HWC')
# 运行一下之后,在tensorboard中查看logs就ok
for i in range(100):
w.add_scalar("y=2x",2*i,i)
pass
w.close()
3.torch中学习使用tensorboard
最新推荐文章于 2024-04-12 18:55:59 发布