PyTorch_3_TensorBoard的使用

运行TensorBoard的指令
tensorboard –logdir=logs

from torch.utils.tensorboard import SummaryWriter
from PIL import Image
import numpy as np

writer = SummaryWriter("logs")

image_path = "practice/train/ants_image/5650366_e22b7e1065.jpg"
img_PIL = Image.open(image_path)
img_array = np.array(img_PIL)

print(type(img_array))
print(img_array.shape)

writer.add_image("ant", img_array, 2, dataformats='HWC')
# y = 2x

for i in range(100):
    writer.add_scalar("y=2x", 3*i, i)

writer.close()