Mac 实现keras网络模型可视化【conda】

最重要的事情:安装顺序不能错,graphviz包/graphviz应用/pydot/pydot_ng

否则pydot安装时没有graphviz,会出现问题。

1.首先你得Mac得有homebrew【建议安装,很方便】:

ruby -e "$(curl -fsSL https://raw.githubusercontent.com/Homebrew/install/master/install)"

2.安装graphviz,打开conda的terminal:

conda install graphviz

3.安装graphviz应用【也在conda terminal里安装,否则pydot找不到路径,conda是一个独立的环境】:

brew install graphviz

4.安装pydot和pydot_ng

pip install pydot pydot_ng

5.进行可视化,这里由于我用的是cGAN,并且编了一个网络的类,所以是这样的

from keras.utils.vis_utils import plot_model
plot_model(self.generator, to_file='modelG.png', show_shapes=True, show_layer_names=False)
plot_model(self.discriminator, to_file='modelD.png', show_shapes=True, show_layer_names=False)
plot_model(self.combined, to_file='modelComb.png', show_shapes=True, show_layer_names=False)

效果

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值