"""
    SummaryWriter：记录步骤
"""
from torch.utils.tensorboard import SummaryWriter
import numpy as np
from PIL import Image

# 记录存储的位置
writer = SummaryWriter("logs/003")

image_path = "dataset/train/ants/0013035.jpg"

img_PIL = Image.open(image_path)
print(type(img_PIL))
# 将 PIL类型 转为 np类型
img_array = np.array(img_PIL)
print(type(img_array))

# 输出行数（高）、列数（宽）、层树（通道数）
print(img_array.shape)

# 记录图片。HWC：高度*宽度*通道数
writer.add_image("记录图片：", img_array, 1, dataformats="HWC")

for i in range(100):
    # 记录坐标对
    writer.add_scalar("坐标对：y=2x", 2 * i, i)

writer.close()