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,他就会在下方独立展示新的。
标签:--------,tensor,img,image,add,numpy,图片 From: https://www.cnblogs.com/ar-boke/p/17068405.html