可视化工具Visdom总结

1. 基本使用方法

# ---- 按特定的port打开visdom ----
visdom -port 8097 -enable_login

2. Python用法

import visdom
import torch
import numpy as np
# ------ 指定environment ------
vis = visdom.Visdom(env='test')

# --------- vis.line ---------
x = torch.arange(0, 10, 0.01)
y = torch.sin(x)
vis.line(X=x, Y=y, win='sinx', opts={'title': 'y=sin(x)'})

# ------ update中的append ------
for i in range(10):
    x = torch.Tensor([i])
    y = x**2
    vis.line(X=x,Y=y, win='sinx', update='append')

# ----------- image -----------
vis.image(np.random.randn(3, 64, 64), win='image')

# ----------- images -----------
vis.images(np.random.randn(32, 3, 64, 64), nrow=8, win='images')

# ----------- heatmap -----------
viz.heatmap(
    X=np.outer(np.arange(1, 6), np.arange(1, 11)),
    opts=dict(
        columnnames=['a', 'b', 'c', 'd', 'e', 'f', 'g', 'h', 'i', 'j'],
        rownames=['y1', 'y2', 'y3', 'y4', 'y5'],
        colormap='Electric',
    )
)