1. Keras中提供了可视化的Tool,将网络结构保存为图片即可查看
2. 加载模型打印可视化结构
from keras.models import load_model #用于加载模型
from keras.utils import plot_model # 用于可视化
model_path="D:/FacialExpressionInteraction/face_classification-master/trained_models/fer2013_mini_XCEPTION.119-0.65.hdf5"
model=load_model(model_path, compile=False) //成功加载了model
plot_model(model, to_file='model.png') //将模型model的结构保存到model.png图片
其实也就是plot_model()
方法而已
BUG1:该方法可能报错如下:
('Failed to import pydot. You must `pip install pydot` and install graphviz (https://graphviz.gitlab.io/download/), ', 'for `pydotprint` to work.')
windows解决方法:
- 下载适用于graphviz 的可执行包安装
- ❤注意:在安装 graphviz 的时候有个选择是否加入环境变量的选择,选择[为所有人添加],否则后面你需要自己去手动添加
- 在你使用的环境中使用pip或者conda安装graphviz和pydotplus(不要问我为什么不是安装pydot),如下
conda install graphviz conda install pydotplus
参考一:关于Failed to import pydot. You must install pydot and graphviz for pydotprint to work的处理建议
参考二:You must pip install pydot and install graphviz
其他OS:自行百度
BUG2:在jupyter Lab中仍报BUG1的错,在Pycharm正常
尝试在Pycharm中可行