目标:

1、使用Tensorboard

2、图像变换,使用transform

首先先了解SummaryWriter,它能直接将条目写入日志目录中的事件文件,供TensorBoard使用

Summarywriter主要使用到两种方法:add_image()和add_scalar()

Summarywriter.add_scalar()

在Summarywriter.add_scalar()的参数中,global_step对应x轴,scalar_value对应y轴 writer.add_scalar()


#导入Summarywriter
from torch.utils.tensorboard import SummaryWriter

#创建实例,把对应的事件文件存储到logs文件夹下
writer =SummaryWriter("logs")

#主要使用到两种方法
#writer.add_image()

#writer.add_scalar()的参数中,global_step对应x轴,scalar_value对应y轴

#如y=x
for i in range(100):
    #第一个参数是表达式(标签),第二个是y轴,第三个是x轴
    writer.add_scalar("y=2x",2*i,i)
writer.close()

如何打开事件文件

1、terminal

 但是默认为6006端口,但是可能有很多人使用,为了防止冲突,咱们可以指定端口。

 点击链接,如果显示错误,则取logs的路径需要写细一点,取log的路径如下

右击logs,——>copy path—>path from content root

 把logdir的路径换成刚刚的取的路径,如下: 

 点击链接,即可看到结果图:

 当改了函数时,可能会导致以前的函数都画在上面了,此时把logs里的文件删掉,在重新运行就可以了

 

图像的研究 Summarywriter.add_image()

1、在python console下(可以不在这里运行)

执行如下操作 ,查看图片,首先要先准备一张图片,(我的是sdgdf.png)

  

 我们可以转成tensor类型或者numpy类型,tensor类型在下一篇文章,此处转成numpy类型

2、 在代码端


#导入Summarywriter
from torch.utils.tensorboard import SummaryWriter
import numpy as np
from PIL import Image
#创建实例,把对应的事件文件存储到logs文件夹下
writer =SummaryWriter("logs")
image_path = "D:\paper reading\code\learningcode\data\sdgdf.png"
img_PIL = Image.open(image_path)
img_array = np.array(img_PIL)
print(type(img_array))
print(img_array.shape)

#HWC分别是高度、宽度和通道
writer.add_image("test",img_array,1,dataformats='HWC')

#主要使用到两种方法
#writer.add_image()

#writer.add_scalar()的参数中,global_step对应x轴,scalar_value对应y轴


for i in range(100):
    #第一个参数是表达式(标签),第二个是y轴,第三个是x轴
    writer.add_scalar("y=2x",2*i,i)
writer.close()

运行后在,terminal下点开链接

打开链接,可以显示图片

 当想在同一个title下,展示多张图片是,修改下面的内容

 运行后,链接里,展示如下:

也可以一张一张显示,此时修改的内容如下: 

 

结果如下:

Logo

有“AI”的1024 = 2048,欢迎大家加入2048 AI社区

更多推荐