首先要安装conda虚拟环境环境,可以参考这里。
激活虚拟环境并安装以下两个包
conda install pydotplus
conda install graphviz
打开模型文件
from keras.utils.vis_utils import plot_model
model = InceV4.Incetion_v4(input_shape,outpt_shape)#你的机器学习模型实例
plot_model(model,to_file=model.name+'_dataflow.pdf',show_shapes=True,show_layer_names=False)#保持模型结构图
例如模型结构如下