python graphviz可视化

安装:

https://graphviz.org/download/

添加系统变量,重启软件

python测试代码如下:

import torch
from torchviz import make_dot
from torchvision.models import vgg16  # 以 vgg16 为例

x = torch.randn(4, 3, 32, 32)  # 随机生成一个张量
model = vgg16()  # 实例化 vgg16,网络可以改成自己的网络
out = model(x)   # 将 x 输入网络
g = make_dot(out)  # 实例化 make_dot
#g.view()  # 直接在当前路径下保存 pdf 并打开
g.render(filename='netStructure/myNetModel', view=False, format='pdf')  # 保存 pdf 到指定路径不打

运行报错:

graphviz.backend.execute.calledprocesserror: command '['dot', '-kdot', '-tpdf', '-o', 'mynetmodel']'

解决方案:

  1. 打开cmd
  2. 运行:dot -v
  3. 出现问题:There is no layout engine support for “dot”…
  4. 运行:dot -c
  5. 没有提示(成功)
  6. 再次运行dot -v

 

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值