python - 以 LeNet 风格绘制 CNN 架构

Question

我想以 LeNet 风格绘制我的 CNN 模型。我找到了在线模拟器http://alexlenail.me/NN-SVG/LeNet.html但我不知道如何将它与我的模型一起使用。我想用我的模型参数填充站点，但我不知道应该在站点的每个步骤中编写的参数。

任何帮助，将不胜感激。我曾多次尝试以 LeNet 风格绘制我的模型，但我做不到。

模型：

    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.layers.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.layers.MaxPooling2D(pool_size=(2, 2), padding='same'))
    model.add(BatchNormalization())
     
    model.add(Dense(32, input_dim=32,kernel_regularizer=regularizers.l2(0.1)))
    model.add(keras.layers.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'))

    opt_adam = keras.optimizers.Adam(lr=0.0001, beta_1=0.9, beta_2=0.999, epsilon=1e-08, decay=0.0)
    model.compile(loss=categorical_focal_loss(), optimizer=opt_adam, metrics=['accuracy'])

标签： pythonkerasmodelarchitectureconv-neural-network