tensorboard的使用

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

writer=SummaryWriter("logs")
#这里使用相对地址，或者img_path="hymenoptera_data\\train\\ants\\0013035.jpg"
img_path="hymenoptera_data/train/ants/0013035.jpg"
#获取图片
img=Image.open(img_path)
#转变格式
img_array=np.array(img)
#将其加入到tensorboard中
writer.add_image("img_narry",img_array,dataformats="HWC")
#注意这个很重要
writer.close()

运行步骤：首先右击运行该文件后，会生成一个log.文件夹，并生成对应的记录文件，然后在终端输入，点击该网址即可

tensorboard --logdir=logs

运行结果