Как мне отобразить эту архитектуру CNN в виде диаграммы? - PullRequest
Новая
       94

Как мне отобразить эту архитектуру CNN в виде диаграммы?

0 голосов
/ 16 июня 2020

Я работаю над проектом и хочу визуализировать шаги, используемые в этой сверточной нейронной сети. До сих пор я рассматривал CNN в основном как черный ящик, поэтому я не совсем уверен, как он работает (я сам не писал его).

CNN записывается на Python с использованием Tensorflow : 

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'))

Я пробовал использовать такие инструменты, как этот , но мне кажется, что диаграмма становится слишком большой.

...