Visdom教程(2)：实时展示图片

这是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)