from torch.utils.tensorboard import SummaryWriter
import numpy as np
writer = SummaryWriter("logs")
#介绍常使用的两个方法:
# writer.add_image()
# writer.add_scalar()
#演示1 :
# for i in range(100):
# writer.add_scalar("y=x",i,i)
# writer.close()
#运行结束后终端打开 tensorboard --logdir=logs --port=6007 (端口默认6006 可选)
#演示2 :
img_path = "dataset/train/dog/dog.jpg"
img_PIL = Image.open(img_path)
print(type(img_PIL)) #查看是否满足writer.add_image()的类型,如果使用opencv读取的直接是numpy型
img_array = np.array(img_PIL) #使用numoy转换
print(type(img_array)) #查看类型
print(img_array.shape) #查看形状
writer.add_image("test",img_array, 1, dataformats='HWC') #dataformats='HWC'指定数据类型
writer.close()
TensorBoard的使用
最新推荐文章于 2024-08-30 12:42:13 发布
![](https://img-home.csdnimg.cn/images/20240711042549.png)