如何使用pytorchviz和Netron可視化pytorch網(wǎng)絡(luò)結(jié)構(gòu)

今天就跟大家聊聊有關(guān)如何使用pytorchviz和Netron可視化pytorch網(wǎng)絡(luò)結(jié)構(gòu)，可能很多人都不太了解，為了讓大家更加了解，小編給大家總結(jié)了以下內(nèi)容，希望大家根據(jù)這篇文章可以有所收獲。

創(chuàng)新互聯(lián)建站是專業(yè)的茫崖網(wǎng)站建設(shè)公司，茫崖接單;提供網(wǎng)站設(shè)計制作、成都做網(wǎng)站,網(wǎng)頁設(shè)計,網(wǎng)站設(shè)計,建網(wǎng)站,PHP網(wǎng)站建設(shè)等專業(yè)做網(wǎng)站服務(wù);采用PHP框架,可快速的進(jìn)行茫崖網(wǎng)站開發(fā)網(wǎng)頁制作和功能擴(kuò)展;專業(yè)做搜索引擎喜愛的網(wǎng)站,專業(yè)的做網(wǎng)站團(tuán)隊,希望更多企業(yè)前來合作!

一使用pytorchviz可視化

安裝依賴和pytorchviz

pip install graphviz
pip install tochviz （或pip install git+https://github.com/szagoruyko/pytorchviz）

Graphviz 是 AT&T 開發(fā)的一款開源的圖形可視化軟件，可以根據(jù)dot腳本語言中繪制的無向圖（顯示了對象間最簡單的關(guān)系）畫出直觀的樹形圖。
Graphviz在Windows中的安裝需要下載Release包，并配置環(huán)境變量，否則會報錯：

graphviz.backend.ExecutableNotFound: failed to execute [‘dot’, ‘-Tpng’, ‘-O’, ‘tmp’], make sure the Graphviz executables are on your systems’ PATH

Graphviz下載地址 https://graphviz.gitlab.io/_pages/Download/Download_windows.html

如何使用pytorchviz和Netron可視化pytorch網(wǎng)絡(luò)結(jié)構(gòu)

下載之后解壓出來是一個“release”文件夾，把“release\bin”目錄添加到系統(tǒng)環(huán)境變量，之后在終端中輸入“dot -V”，顯示以下信息表示Graphviz配置成功：

torchviz可視化torch網(wǎng)絡(luò)結(jié)構(gòu)

# Created by 牧野 CSDN
import torch
from torch import nn
from torchviz import make_dot, make_dot_from_trace
model = nn.Sequential()
model.add_module('W0', nn.Linear(8, 16))
model.add_module('tanh', nn.Tanh())
model.add_module('W1', nn.Linear(16, 1))
x = torch.randn(1,8)
vis_graph = make_dot(model(x), params=dict(model.named_parameters()))
vis_graph.view() # 會在當(dāng)前目錄下保存一個“Digraph.gv.pdf”文件，并在默認(rèn)瀏覽器中打開
with torch.onnx.set_training(model, False):
trace, _ = torch.jit.get_trace_graph(model, args=(x,))
make_dot_from_trace(trace)

調(diào)用“make_dot”方法創(chuàng)建一個dot對象，使用“view”方法顯示出來。

pytorch2.2和1.3版本中使用“torch.jit.get_trace_graph”可能會報錯，1.1版本ok。

AttributeError: 'torch._C.Value' object has no attribute 'uniqueName'

可視化結(jié)果：

如何使用pytorchviz和Netron可視化pytorch網(wǎng)絡(luò)結(jié)構(gòu)

二使用Netron可視化

Netron開源地址： https://github.com/lutzroeder/Netron
Netron的開發(fā)者是Lutz Roeder，一位來自微軟Visual Studio團(tuán)隊的帥哥：

Netron是一款支持離線查看“各種”神經(jīng)網(wǎng)絡(luò)框架的模型可視化神器，其中的“各種”包括：

ONNX (.onnx, .pb, .pbtxt)
Keras (.h6, .keras)
Core ML (.mlmodel)
Caffe (.caffemodel, .prototxt)
Caffe2 (predict_net.pb, predict_net.pbtxt)
MXNet (.model, -symbol.json)
NCNN (.param)
TensorFlow Lite (.tflite)
TorchScript (.pt, .pth)
PyTorch (.pt, .pth)
Torch (.t7)
Arm NN (.armnn)
BigDL (.bigdl, .model)
Chainer, (.npz, .h6)
CNTK (.model, .cntk)
Deeplearning4j (.zip)
Darknet (.cfg)
ML.NET (.zip)
MNN (.mnn)
OpenVINO (.xml)
PaddlePaddle (.zip, __model__)
scikit-learn (.pkl)
TensorFlow.js (model.json, .pb)
TensorFlow (.pb, .meta, .pbtxt)

嗯，夠多了。

Netron使用很簡單，作者提供了各個平臺的安裝包，安裝之后打開，把保存的模型文件拖入就可以了。
還以上邊的模型為例，先把pytorch模型保存出來：

import torch
from torch import nn
from torchviz import make_dot, make_dot_from_trace
model = nn.Sequential()
model.add_module('W0', nn.Linear(8, 16))
model.add_module('tanh', nn.Tanh())
model.add_module('W1', nn.Linear(16, 1))
torch.save(model, 'model.pth') # 保存模型

之后用Netron打開保存的“model.pth”：

如何使用pytorchviz和Netron可視化pytorch網(wǎng)絡(luò)結(jié)構(gòu)

網(wǎng)絡(luò)結(jié)構(gòu)很清晰，一目了然，右側(cè)還能顯示操作的進(jìn)一步信息。

看完上述內(nèi)容，你們對如何使用pytorchviz和Netron可視化pytorch網(wǎng)絡(luò)結(jié)構(gòu)有進(jìn)一步的了解嗎？如果還想了解更多知識或者相關(guān)內(nèi)容，請關(guān)注創(chuàng)新互聯(lián)行業(yè)資訊頻道，感謝大家的支持。

新聞標(biāo)題：如何使用pytorchviz和Netron可視化pytorch網(wǎng)絡(luò)結(jié)構(gòu)
轉(zhuǎn)載來于：http://muchs.cn/article40/geopeo.html

成都網(wǎng)站建設(shè)公司_創(chuàng)新互聯(lián)，為您提供網(wǎng)站收錄、網(wǎng)站設(shè)計公司、動態(tài)網(wǎng)站、移動網(wǎng)站建設(shè)、軟件開發(fā)、品牌網(wǎng)站建設(shè)

聲明：本網(wǎng)站發(fā)布的內(nèi)容（圖片、視頻和文字）以用戶投稿、用戶轉(zhuǎn)載內(nèi)容為主，如果涉及侵權(quán)請盡快告知，我們將會在第一時間刪除。文章觀點不代表本網(wǎng)站立場，如需處理請聯(lián)系客服。電話：028-86922220；郵箱：631063699@qq.com。內(nèi)容未經(jīng)允許不得轉(zhuǎn)載，或轉(zhuǎn)載時需注明來源：創(chuàng)新互聯(lián)

猜你還喜歡下面的內(nèi)容

如何使用pytorchviz和Netron可視化pytorch網(wǎng)絡(luò)結(jié)構(gòu)

一 使用pytorchviz可視化

安裝依賴和pytorchviz

torchviz可視化torch網(wǎng)絡(luò)結(jié)構(gòu)

二 使用Netron可視化

一使用pytorchviz可視化

二使用Netron可視化