Have fun with sci.dog

keras绘制神经网络结构图

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安装,要到官方下载安装

Download | Graphviz

下载最新的5.-0的x64版本,一路next,最后记得把环境变量添加到电脑的勾给打上。

最后,运行绘图代码,成功

赞(0)
未经允许不得转载:SciDog » keras绘制神经网络结构图

评论 抢沙发