Transorboard

`SummaryWriter` 类提供了一个高级 API，用于在给定目录中创建事件文件，并向其中添加摘要和事件。 该类异步更新文件内容。 这允许训练程序调用方法以直接从训练循环将数据添加到文件中，而不会减慢训练速度。pytorch的可视化工具

安装

pip install --upgrade pip setuptools wheel
pip install opencv-python==4.5.4.60 -i https://pypi.tuna.tsinghua.edu.cn/simple

 如果出现LooseVersion = distutils.version.LooseVersion tensorboard错误，确保版本大于1.15，注释掉几行

import tensorboard
from setuptools import distutils

# LooseVersion = distutils.version.LooseVersion

# if not hasattr(tensorboard, '__version__') or LooseVersion(tensorboard.__version__) < LooseVersion('1.15'):
#     raise ImportError('TensorBoard logging requires TensorBoard version 1.15 or above')

del distutils
#del LooseVersion
del tensorboard

from .writer import FileWriter, SummaryWriter  # noqa: F401
from tensorboard.summary.writer.record_writer import RecordWriter  # noqa: F401

SummaryWriter用于在给定目录中创建事件文件，并向其中添加摘要和事件，更新文件内容。、

PIL对象

array阵列

numpy转化成数组

H:高度 W:宽度 C:通道

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

writer = SummaryWriter("logs")
image_path = "dataset/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("test", img_array, 2, dataformats='HWC')
# y=2x
for i in range(100):
    writer.add_scalar("y=2x", 3 * i, i)
writer.close()