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