Keras实践:模型可视化
Keras实践:模型可视化
安装Graphviz
官方网址为:http://www.graphviz.org/。我使用的是mac系统,所以我分享一下我使用时遇到的坑。
Mac安装时在终端中执行:
brew install graphviz
若卡在Updating Homebrew....,需要取消brew的自动更新:
安装PyDot
加载模型并生成结构图
from keras.models import load_model from keras.utils.vis_utils import plot_model mymodel = load_model("/Users/apple/PycharmProjects/MyDjango/myweb/CNNModel.hdf5") plot_model(emotion_classifier,'model.png')