如何将这种CNN架构显示为图表?

时间:2020-06-16 14:16:26

标签: python tensorflow machine-learning keras conv-neural-network

我正在做一个项目,我想可视化此卷积神经网络中使用的步骤。到目前为止,我一直将CNN视为一个黑匣子,因此,我不确定它的工作原理(我自己也没有写过)。

CNN使用Tensorflow用Python编写:

model = Sequential()

model.add(Conv2D(32, kernel_size=(3, 3), activation='relu', input_shape=(48,48,1)))
model.add(Conv2D(64, kernel_size=(3, 3), activation='relu'))
model.add(MaxPooling2D(pool_size=(2, 2)))
model.add(Dropout(0.25))

model.add(Conv2D(128, kernel_size=(3, 3), activation='relu'))
model.add(MaxPooling2D(pool_size=(2, 2)))
model.add(Conv2D(128, kernel_size=(3, 3), activation='relu'))
model.add(MaxPooling2D(pool_size=(2, 2)))
model.add(Dropout(0.25))

model.add(Flatten())
model.add(Dense(1024, activation='relu'))
model.add(Dropout(0.5))
model.add(Dense(7, activation='softmax'))

我曾尝试使用this one之类的工具,但我觉得该图越来越大。

0 个答案:

没有答案