8、TensorBoard的使用(二)--------add_image()常用来观察训练结果

1、add_image()

参数:add_image(self,tag,img_tensor,global_step=None,walltime=None,dataformats='CHW‘)
tag:(string)Data identifier数据标签;  img_tensor:(必须是torch.Tensor类型,或者numpy.array/string/blobname类型);  global_step:(int)训练的步骤; '''
***注意:
读取图片的时候是要先获取图片的地址的,然后再用Image库的open函数打开图片,但是,图片的数据类型需要做转换,必须满足img_tensor要求。

 

****两种图片类型转换方法:

1、转为numpy:

   利用opencv读取图片,获得numpy型图片数据

   利用 numpy.array(), 对PIL图片进行转换:

    导入numpy,然后利用numpy.array()函数。

 *****要注意:图像数据的shape形状,尺寸:

  可以使用.shape来查看图像的尺寸  

  默认的是(3,h,w)3个通道,h高度,w宽度。如果通道数在后面,比如(h,w,3)这种形式的话需要修改参数dataformats:CHW,HWC。

 

 

 然后打开logs事件文件查看就会有图片显示。

 

 如果想单独展示,可以更改一下tag,他就会在下方独立展示新的。

 

 

 

 

posted @ 2023-02-02 21:46  bokeAR  阅读(363)  评论(0)    收藏  举报