Как мне определить размер ввода для слоев, следующих за начальным слоем в CNN? ⇐ Python

[Anonymous]

Я работаю над модулем 5 CS50AI, и это код из части лекции, посвященной распознаванию чисел. Если бы я хотел добавить еще один сверточный слой после максимального объединения, как бы я определил входную форму? Будет ли это IMG_WIDTH, IMG_HEIGHT, 3 или я разделю IMG_WIDTH и IMG_HEIGHT на 2 из-за максимального пула?
Аналогично, как определяется количество узлов в первом плотном слое (128) решил? Это произвольное число, которое я могу решить, или оно основано на чем-то другом?
model = tf.keras.models.Sequential([
tf.keras.layers.Conv2D(
32, (5, 5), activation="relu", input_shape=(IMG_WIDTH, IMG_HEIGHT, 3)
),
tf.keras.layers.MaxPooling2D(pool_size=(2, 2)),
tf.keras.layers.Flatten(),
tf.keras.layers.Dense(128, activation="relu"),
tf.keras.layers.Dropout(0.5),
tf.keras.layers.Dense(NUM_CATEGORIES, activation="softmax")
])

Для справки
IMG_WIDTH = 30
IMG_HEIGHT = 30
NUM_CATEGORIES = 3


Подробнее здесь: https://stackoverflow.com/questions/791 ... l-layer-in