4.TensorBoard的使用(4.4)
4.1 add_scalar()的使用
整体代码
1)首先安装TensorBoard,在终端输入以下代码
2)打开event files的方法。logdir=event files所在文件夹;点击超链接即可打开
# 可通过对参数的设置,避免在同一服务器上运行时和其他人的端口一样
3)在不改标题的情况下改变 y ,会导致图像出错
一种解决方法是在更改后删除所有event files , run 后重复步骤2
结果
4.2 add_image()的使用
1)"Ctrl"+点击函数名获取帮助,该函数适用的图片格式如下
可知所求图片格式不包括在内
2)利用函数 numpy.arry(),对PIL图片格式进行转换。验证格式成功转换
# 以 np 代替 numpy,之后可以用 np 调用
3)帮助文档里,数据格式默认为 'CHW' ,而所求图片不是,若不加以定义则会报错
4)此外,从 PIL 到 numpy,需要在 add_image() 中指定shape中每一个数字/维表示的含义
若改变图片和迭代次数,而不改变标题,则结果为
4.3 总结
利用 TensorBoard 可以直观地看到训练过程中 model 使用的数据,或者在对 model 进行测试时看到每一阶段的输出结果。