visdom 快速啟動

啟動地址：http://localhost:8097/

visdom 的介紹

Visdom是Facebook專為PyTorch開發的實時可視化工具包，其作用相當于TensorFlow中的Tensorboard，靈活高效且界面美觀，下面就一起來學習下如何使用吧！如果想更多了解關于Visdom的使用可以參考官方

首先來欣賞下官方提供的Visdom的可視化界面

Visdom的安裝

• 安裝非常簡易，只需要打開cmd窗口，輸入一下命令即可快速安裝完成

pip install visdom

Visdom 的使用

類似于TensorFlow的TensorBoard，要使用Visdom，就要先在終端開啟監聽命令，根據顯示的網址然后在瀏覽器里輸入：http://localhost:8097 進行登錄，此時如果報錯，別怕，參考以下網站一定能輕松解決（新版visdom已經解決了可以使用pip install --upgrade visdom進行更新即可）：

開啟監聽命令

python -m visdom.server # 或者直接visdom

示例1：實時曲線繪制方法

• 方法是起點+數據點更新

from visdom import Visdom''' 單條追蹤曲線設置 ''' viz = Visdom() # 初始化visdom類 viz.line([0.], ## Y的第一個點坐標[0.], ## X的第一個點坐標win="train loss", ##窗口名稱opts=dict(title='train_loss') ## 圖像標例) #設置起始點 ''' 模型數據 ''' viz.line([1.], ## Y的下一個點坐標[1.], ## X的下一個點坐標win="train loss", ## 窗口名稱 與上個窗口同名表示顯示在同一個表格里update='append' ## 添加到上一個點后面) # Setting up a new session...# 'train loss'

• 界面自動更新如下
  

示例2：多條曲線繪制

• 多條曲線繪制

from visdom import Visdom''' 多條曲線繪制 實際上就是傳入y值時為一個向量 ''' viz = Visdom(env='my_wind') # 注意此時我已經換了新環境 #設置起始點 viz.line([[0.0,0.0]], ## Y的起始點[0.], ## X的起始點win="test loss", ##窗口名稱opts=dict(title='test_loss') ## 圖像標例) ''' 模型數據 ''' viz.line([[1.1,1.5]], ## Y的下一個點[1.], ## X的下一個點win="test loss", ## 窗口名稱update='append' ## 添加到上一個點后面) # 'test loss'

• 大家此時查看需要先切換environment窗口為my才能看到圖像，如圖所示：
  

示例3：圖像顯示

• 值得注意的是，Visdom支持圖像的批量顯示

from visdom import Visdom import numpy as npimage = np.random.randn(6, 3, 200, 300) # 此時batch為6 viz = Visdom(env='my_image') # 注意此時我已經換了新環境 viz.images(image, win='x')# 'x'

• 界面自動更新如下
  

示例4：可視化數據集

• 值得注意的是，Visdom支持圖像的批量顯示

from visdom import Visdom import numpy as np import torch from torchvision import datasets, transforms# 注意數據集路徑 train_loader = torch.utils.data.DataLoader(datasets.MNIST(r'D:\data',train=True,download=False,transform=transforms.Compose([transforms.ToTensor()])),batch_size=128,shuffle=True) sample=next(iter(train_loader)) # 通過迭代器獲取樣本 # sample[0]為樣本數據 sample[1]為類別 nrow=16表示每行顯示16張圖像 viz = Visdom(env='my_visual') # 注意此時我已經換了新環境 viz.images(sample[0],nrow=16,win='mnist',opts=dict(title='mnist'))# 'mnist'

• 界面自動更新如下
  

示例5：訓練

• 下面通過具體的訓練過程通過visdom可視化
• 為了方便顯示Visdom的功能，直接使用自帶的MNist數據進行可視化。

''' 導入庫文件 ''' import torch import torch.nn as nn import torch.nn.functional as F import torch.optim as optim from torchvision import datasets, transforms from visdom import Visdom import numpy as np''' 構建簡單的模型:簡單線性層+Relu函數的多層感知機 ''' class MLP(nn.Module):def __init__(self):super(MLP, self).__init__()self.model = nn.Sequential(nn.Linear(784, 200),nn.ReLU(inplace=True),nn.Linear(200, 200),nn.ReLU(inplace=True),nn.Linear(200, 10),nn.ReLU(inplace=True),)def forward(self, x):x = self.model(x)return xbatch_size = 128 learning_rate = 0.01 epochs = 10# 注意數據集路徑 train_loader = torch.utils.data.DataLoader(datasets.MNIST(r'D:\Users\Administrator\Desktop\PythonDLbasedonPytorch\data', train=True,download=True,transform=transforms.Compose([transforms.ToTensor(),transforms.Normalize((0.1307, ), (0.3081, ))])),batch_size=batch_size,shuffle=True) # 注意數據集路徑 test_loader = torch.utils.data.DataLoader(datasets.MNIST(r'D:\Users\Administrator\Desktop\PythonDLbasedonPytorch\data',train=False,transform=transforms.Compose([transforms.ToTensor(),transforms.Normalize((0.1307, ), (0.3081, ))])),batch_size=batch_size,shuffle=True)# 注意此處初始化visdom類 viz = Visdom() # 繪制起點 viz.line([0.], [0.], win="train loss", opts=dict(title='train_loss')) device = torch.device('cuda:0') net = MLP().to(device) optimizer = optim.SGD(net.parameters(), lr=learning_rate) criteon = nn.CrossEntropyLoss().to(device)for epoch in range(epochs):for batch_idx, (data, target) in enumerate(train_loader):data = data.view(-1, 28 * 28)data, target = data.to(device), target.cuda()logits = net(data)loss = criteon(logits, target)optimizer.zero_grad()loss.backward()# print(w1.grad.norm(), w2.grad.norm())optimizer.step()if batch_idx % 100 == 0:print('Train Epoch: {} [{}/{} ({:.0f}%)]\tLoss: {:.6f}'.format(epoch, batch_idx * len(data), len(train_loader.dataset),100. * batch_idx / len(train_loader), loss.item()))test_loss = 0correct = 0for data, target in test_loader:data = data.view(-1, 28 * 28)data, target = data.to(device), target.cuda()logits = net(data)test_loss += criteon(logits, target).item()pred = logits.argmax(dim=1)correct += pred.eq(target).float().sum().item()test_loss /= len(test_loader.dataset)# 繪制epoch以及對應的測試集損失lossviz.line([test_loss], [epoch], win="train loss", update='append')print('\nTest set: Average loss: {:.4f}, Accuracy: {}/{} ({:.0f}%)\n'.format(test_loss, correct, len(test_loader.dataset),100. * correct / len(test_loader.dataset)))

loss曲線如圖所示

visdom 基本可視化函數

• 具體使用方法仍然可以參考上述網站，限于篇幅，這里主要列舉最常用的line函數以及image函數的使用方法

visdom基本可視化函數

- vis.image : 圖片 - vis.line: 曲線 - vis.images : 圖片列表 - vis.text : 抽象HTML 輸出文字 - vis.properties : 屬性網格 - vis.audio : 音頻 - vis.video : 視頻 - vis.svg : SVG對象 - vis.matplot : matplotlib圖 - vis.save : 序列化狀態服務端

上述函數參數

• 注意opt的參數都可以用python字典的格式傳入，大家可以參考下方使用方法

- opts.title : 圖標題 - opts.width : 圖寬 - opts.height : 圖高 - opts.showlegend : 顯示圖例 (true or false) - opts.xtype : x軸的類型 ('linear' or 'log') - opts.xlabel : x軸的標簽 - opts.xtick : 顯示x軸上的刻度 (boolean) - opts.xtickmin : 指定x軸上的第一個刻度 (number) - opts.xtickmax : 指定x軸上的最后一個刻度 (number) - opts.xtickvals : x軸上刻度的位置(table of numbers) - opts.xticklabels : 在x軸上標記標簽 (table of strings) - opts.xtickstep : x軸上刻度之間的距離 (number) - opts.xtickfont :x軸標簽的字體 (dict of font information) - opts.ytype : type of y-axis ('linear' or 'log') - opts.ylabel : label of y-axis - opts.ytick : show ticks on y-axis (boolean) - opts.ytickmin : first tick on y-axis (number) - opts.ytickmax : last tick on y-axis (number) - opts.ytickvals : locations of ticks on y-axis (table of numbers) - opts.yticklabels : ticks labels on y-axis (table of strings) - opts.ytickstep : distances between ticks on y-axis (number) - opts.ytickfont : font for y-axis labels (dict of font information) - opts.marginleft : 左邊框 (in pixels) - opts.marginright :右邊框 (in pixels) - opts.margintop : 上邊框 (in pixels) - opts.marginbottom: 下邊框 (in pixels) - opts.lagent=['']： 顯示圖標

參考鏈接

輕松學 Pytorch–Visdom 可視化

GitHub 地址

PyTorch的遠程可視化神器visdom

總結

以上是生活随笔為你收集整理的PyTorch 训练可视化教程 visdom的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。