Pytorch——SummaryWriter.add_image()的使用

from torch.utils.tensorboard import SummaryWriter

用法

image_path="data/train/ants_image/6240329_72c01e663e.jpg"#图片路径
#读取图片的两种方式:
img_PIL=Image.open(image_path) #方式一
img_PIL=cv2.imread(image_path) #方式二
img_array=np.array(img_PIL)
writer=SummaryWriter("logs")  #文件夹“logs”
writer.add_image('ant',img_array,1, dataformats='HWC')
#参数("显示图像的窗口名",显示图片的名字,1表示一张图片,两张图片为2...
writer.close()#关闭

显示 :

1、在控制台输入:

tensorboard --logdir=logs#=文件名  

 
  2、 修改端口号方式:

 tensorboard --logdir=logs --port=6007#自己制定显示的端口号

 

 结果:


版权声明:本文为m0_62727050原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接和本声明。