TensorBoard的使用，add_image()的使用。

在TensorBoard中，add_image()函数用于将图像数据添加到可视化中。它可以用于显示模型输入、输出、中间特征图等图像数据，以帮助开发者理解模型的运行情况。

add_image()的用法：

使用ctrl点击add_image()

注意：图片类型要求为

torch.Tensor, numpy.ndarray, 或者 string/blobname型

故要将图片类型进行转换

from torch.utils.tensorboard import SummaryWriter
from PIL import Image
import numpy as np
img_path='../data/train/ants/506249802_207cd979b4.jpg' #获取图片的地址
img_PIL=Image.open(img_path)  #打开该图片，图片属于PIL型
img_array=np.array(img_PIL)  #将PIL型图片使用numpy转换为numpy型

以上操作基于已安装好numpy包

至此，图片的类型已转换为所需要的类型。

输入以下代码，使用TensorBoard

from torch.utils.tensorboard import SummaryWriter
from PIL import Image
import numpy as np
img_path='../data/train/ants/506249802_207cd979b4.jpg' #获取图片的地址
img_PIL=Image.open(img_path)  #打开该图片，图片属于PIL型
img_array=np.array(img_PIL)  #将PIL型图片使用numpy转换为numpy型

writer=SummaryWriter('logs')

writer.add_image('test',img_array,1,dataformats="HWC")

writer.close()

在控制台终端指定端口输出：

输出结果为：

为了使用add_image()函数，需要确保已经安装了TensorFlow和TensorBoard，并且将相关的日志数据写入到TensorBoard的日志文件中（使用tf.summary.create_file_writer()和with summary_writer.as_default()）。