深度学习简单画图

深度学习简单画图

画树形图———>使用Pytorch的torchviz画图

安装过程参考博客

graphviz.backend.ExecutableNotFound: failed to execute ‘dot‘Python使用Graphviz绘图报错解决_whtli的博客-优快云博客

pytorch 网络可视化(二)：graphviz + torchviz_听 风、的博客-优快云博客_pytorch torchviz 可视化 主要看的是这个安装torchviz

Pytorch的网络可视化_翊夜的博客-优快云博客 对比了torchviz和netron画出的图

安装步骤

1. 官网下载 https://www.graphviz.org/download/

2. 安装完后确定环境变量中有

   

3. 进入anconda的相应虚拟环境下

   pip install graphviz torchviz
   

4.把pytorch项目全部关闭，重启pytorch

测试代码

import torch
import torch.nn as nn
import torchviz


class CNNModel(nn.Module):
    def __init__(self, input_n,hidden_n,ouput_n):
        super(CNNModel, self).__init__()

        self.layer1 = nn.Sequential(nn.Linear(input_n, hidden_n), nn.Sigmoid())
        self.layer2 = nn.Sequential(nn.Linear(hidden_n, ouput_n), nn.Sigmoid())

    def forward(self, x):
        x1 = self.layer1(x)
        x2 = self.layer2(x1)
        return x2

model = CNNModel(2,5,1)
cases = torch.tensor([[0, 0],
                          [0, 1],
                          [1, 0],
                          [1, 1]]).float()
bp=model(cases)
g = torchviz.make_dot(bp)
g.view(filename="hello")        #直接在当前路径下保存并打开
# g.render(filename='netStructure/myNetModel', view=False, format='pdf')  # 保存 pdf 到指定路径且不立刻打开

效果图如下

技巧

把生成的图用wps的pdf打开后，可以修改文字

画层次图

NN SVG (alexlenail.me)