Python 如何绘制keras CNN架构？

我想用我的代码绘制Keras CNN架构。画那个模型有什么好主意吗

任何帮助都将不胜感激

提前谢谢

代码：

import keras
from keras.models import Sequential
from keras.layers import Model, Conv2D, MaxPooling2D, Flatten, Dense,BatchNormalization,Dropout
from pptx_util import save_model_to_pptx
from matplotlib_util import save_model_to_file
input_shape=(33,3840,1)
model=sequential()
#c1
model.add(Conv2D(16,(5,5),strides=(2,2),padding='same',activation='relu',input_shape=input_shape))
model.add(keras.layer.MaxPooling2D(pool_size=(2,2),padding='same'))
model.add(BatchNormalization)
#c2
model.add(Conv2D(32,(3,3),strides=(1,1),padding='same',activation='relu'))
model.add(keras.layer.MaxPooling2D(pool_size=(2,2),padding='same'))
model.add(BatchNormalization)
model.add(Dense(32,input_dim=32,kernel_regularizer=regularizer.l2(0.1)))
model.add(keras.layer.Dropout(0.6))
model.add(Flatten())
model.add(Dropout(0.6))
model.add(Dense(256,activation='sigmoid'))
model.add(Dropout(0.6))
model.add(Dense(2,activation='softmax'))
# save as svg file
model.save_fig("example.svg")
# save as pptx file
save_model_to_pptx(model, "example.pptx")
# save via matplotlib
save_model_to_file(model, "example.pdf")

---

据我所知，keras.utils有一个名为

plot\u model（）

你试过这个吗

tf.keras.utils.plot_model(
    model,
    to_file="model.png",
    show_shapes=False,
    show_dtype=False,
    show_layer_names=True,
    rankdir="TB",
    expand_nested=False,
    dpi=96,
)

---

非常感谢你，我很珍惜你的答复。我想用列奈风格画我的建筑。有什么可以帮我的吗？？