keras绘制网络结构图

keras可以方便画出网络结构图,但在使用过程中有两个库需要先安装 pydot 和graphviz

os: unbantu

pydot使用pip很容易安装
graphviz直接用pip装没问题,但是使用时候遇到环境变量问题。这儿需要将graphviz/bin添加到环境变量,但是我装完发现graphviz文件下无bin文件。。。

另寻他路,apt-get install graphviz直接安装成功并且使用没问题了~

import os
os.environ["CUDA_VISIBLE_DEVICES"] = "  "
from keras.applications.resnet50 import ResNet50
model = ResNet50(weights='imagenet')
from keras.utils import plot_model
plot_model(model, to_file='model.png')
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值