安装
pip install pydot
下载 graphviz.msi 安装并且添加环境变量
使用
from keras.utils.vis_utils import plot_model
import matplotlib.pyplot as plt
# TB：从上往下 LR：左右
plot_model(model,to_file="model.png",show_shapes=True,show_layer_names=True,rankdir='TB')
plt.figure(figsize=(30,30))
img = plt.imread(&q