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#自己制定显示的端口号

 

 结果：