这是Visdom教程的第二篇文章。主要讲了如何使用Visdom在本地查看服务器上的一个图片。
我们常常使用Visdom实时展示模型训练时刻的图像输出。Visdom支持展示numpy数组形式的若干张图片。展示图片用到了如下两个函数
- visdom.image(img, win)
- visdom.images(imgs, win)
第一个函数在win指定的窗口,展示单张图片,而第二个函数则用来展示一系列图片。
- img是一个表示图片的numpy数组,形状(3,width,height)或者(width,height)形状
- imgs是表示一系列图片的numpy数组,形状为(batch, 3, width, height)。
- win表示窗口id,应该可以为任意类型。如果不指定win,服务端将自动新建一个窗口放图片,然后你的浏览器中将堆满密密麻麻的图片窗口。
下面是一个例子——
import numpy as np import time import visdom import numpy as np vis = visdom.Visdom(env='example') while True: img = np.random.randn(3,64,64) vis.image(img, win='single_image') imgs = np.random.randn(10,3,64,64) vis.images(imgs, win='multiple_images') time.sleep(1)