# -*- coding: utf-8 -*-
"""
Created on %(date)s
@Author : %(QuanWang)s
quan.wang.100@strath.ac.uk
"""
from torch.utils.tensorboard import SummaryWriter
import numpy as np
from PIL import Image
writer=SummaryWriter('2021_11_11_log')
image_path="C:/Users/QuanWang/Desktop/2021_Otc_Pytorch_learning_Quan/lesson14_tensorboard/数据集/hymenoptera_data/train/ants/0013035.jpg"
img_PIL= Image.open(image_path)
img_array=np.array(img_PIL)
writer.add_image('test',img_array,1, dataformats='HWC')
for i in range(100):
writer.add_scalar('y=2x', 3*i, i)
writer.close()
image.png
image.png