2021-11-12 tensorboard using

# -*- coding: utf-8 -*-
"""
Created on %(date)s

@Author : %(QuanWang)s
          [email protected]
"""

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

你可能感兴趣的:(2021-11-12 tensorboard using)