Tensorboard(二)：add_image()

一、前期工作：数据集下载

• 将数据集下载好后，解压并放入项目文件夹中

二、函数介绍

add_image() 函数说明

三、运行过程

代码基于上一篇笔记(2条消息) Tensorboard的使用（一）writer.add_scalar()，SummaryWriter()_rensweet的博客-CSDN博客修改

假设取数据集第一张图片，复制图片的相对地址 

 

  运行以下代码报错

• 报错1：Assertion failed

原因：ndarry：

但add_image() 默认为[c,h,w]

•  报错2：Empty suite

未用test命名类或文件，代码正确

后来发现我一直点的是pycharm右上角的运行，只要右键-->run即可

 

按上述过程修改代码运行后， 在terminal中输入代码：

 tensorboard --logdir="E:\PycharmProjects\runs\flower_experiment"

点击蓝色链接，出现以下界面

补：工具台操作

控制台查看数据类型

图：控制台查看数据类型代码

 运行后在右侧可查看结果

 图：数据类型

想更直观的话可以再输入：print(type(img))

 可以看出这样得到的数据类型不符合add_image()所要求的

在控制台输入下列代码，将数据转化为ndarry格式

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

注：如果出现路径错误，可以试一试使用绝对路径