第一次使用需要加入

pip install tensorboard

 安装后，就可以使用tensorboard；

from torch.utils.tensorboard import SummaryWriter

writer=SummaryWriter("logs")

# writer.add_image()
for i in range(100):
    writer.add_scalar("y=x",i,i)

writer.close()

简单画一个y=x图像模拟模型训练时候的损失函数的图像。在终端输入下方指令，点击连接即可查看图像。

tensorboard --logdir=logs --port=6007

img=cv2.imread('./data/train/ants/0013035.jpg')
print(type(img))
writer.add_image('test',img,0,dataformats='HWC')

 注意问题：pillow的版本要求在10.0以下，否则会如下报错：module 'PIL.Image' has no attribute 'ANTIALIAS'