Это из текста Дэвида «Генераторное глубокое обучение». Фостер

Код: Выделить всё
input_layer = layers.Input(shape=(32,32,3))
conv_layer_1 = layers.Conv2D(
filters = 10
, kernel_size = (4,4)
, strides = 2
, padding = 'same'
)(input_layer)
conv_layer_2 = layers.Conv2D(
filters = 20
, kernel_size = (3,3)
, strides = 2
, padding = 'same'
)(conv_layer_1)
flatten_layer = layers.Flatten()(conv_layer_2)
output_layer = layers.Dense(units=10, activation = 'softmax')
model = models.Model(input_layer, выходной_слой)
- Слой (тип) Форма вывода Param #
Conv2D (нет, 16) , 16, 10) 490
Conv2D (нет, 8, 8, 20) 1820
Сглаживание (нет, 1280) 0Плотный (Нет, 10) 12 810
Подробнее здесь: https://stackoverflow.com/questions/793 ... -conv2d-la