P8 TensorBoard的使用（二）

#P8 TensorBoard的使用（二）


pip install opencv-python#安装opencv


image_path = "data/train/ants_image/0013035.jpg"
from PIL import Image
img = Image.open(image_path)
print(type(img))


import numpy as np
img_array = np.array(img)
print(type(img_array))



pip install tensorboard#安装tensorboard

tensorboard --logdir=logs#logdir=事件文件所在文件夹名
tensorboard --logdir=logs --port=6007#port=端口名


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

writer = SummaryWriter("logs")
image_path = "data/val/ants/17081114_79b9a27724.jpg"
img_PIL = Image.open(image_path)
img_array = np.array(img_PIL)
print(type(img_array))
print(img_array.shape)

writer.add_image("train", img_array, 1, dataformats='HWC')#从PIL到numpy，需要在add image()中指定shape中每一个数字/维表示的含义,H高度,W宽度,C通道

# y = 2x

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

writer.close()