可视化工具visdom

服务端启动方法

python -m visdom.server
# 后台启动
nohup python -m visdom.server &

• 使用 cat 查看 nohup 日志
• 使用 jobs 查看进程 pid
• 使用 kill 结束进程

程序使用方法

import visdom
import numpy as np
from time import strftime

class Visualizer(object):
    def __init__(self, env="default", **kwargs):
        self.vis = visdom.Visdom(env=env, use_incoming_socket=False, **kwargs)
        self.index = {}
        self.log_text = ""
    
    def reinit(self, env="default", **kwargs):
        self.vis = visdom.Visdom(env=env, **kwargs)
        return self
    
    def plot_many(self, d):
        for i in d.items():
            self.plot(*i)
    
    def img_many(self, d):
        for i in d.items():
            self.img(*i)
    
    def plot(self, name, y, **kwargs):
        x = self.index.get(name, 0)
        self.vis.line(Y=np.array([y]),
                      X=np.array([x]),
                      win=name,
                      opts=dict(title=name),
                      update=None if x == 0 else "append",
                      **kwargs
                      )
        self.index[name] = x + 1
    
    def img(self, name, img_, **kwargs):
        self.vis.images(img_.cpu().numpy(),
                        win=name,
                        opts=dict(title=name),
                        **kwargs
                        )
    
    def log(self, info, win="log_text"):
        now = strftime("%m%d_%H%M%S")
        self.log_text += f"[{now}]{info}<br/>"
        self.vis.text(self.log_text, win)
    
    def __getattr__(self, item):
        return getattr(self.vis, item)