keras神经网络结构存储在一个json文件里,如图所示
要可视化这个模型,根据网络教程,写了这么个简单脚本
from keras.models import model_from_json
from keras.utils import plot_model
from keras_visualizer import visualizer
from sqlalchemy import false, true
json_file = open('model/model.json', 'r')
loaded_model_json = json_file.read()
json_file.close()
loaded_model = model_from_json(loaded_model_json)
plot_model(loaded_model,to_file='net.png',show_shapes=true)
visualizer(loaded_model,filename='net_view.png',format='png',view=false)
运行提示,缺少pydot和grahpviz
ImportError: You must install pydot (`pip install pydot`) and install graphviz (see instructions at https://graphviz.gitlab.io/download/) for plot_model/model_to_dot to work.
pip安装这两个包后,错误依旧,经过查找,发现grahpviz不能通过pip安装,要到官方下载安装
下载最新的5.-0的x64版本,一路next,最后记得把环境变量添加到电脑的勾给打上。
最后,运行绘图代码,成功