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')

你可能感兴趣的:(问题解决方案)