Как рассчитать выходные размеры после слоя свертки в файле конфигурации?

Question

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

Большинство вещей, на которые я уже смотрел, были не очень ясными и краткими. Как я должен рассчитать размеры через каждый слой? Ниже приведен фрагмент файла конфигурации, который будет проанализирован.

# (3, 640, 640)
[convolutional]
batch_normalize=1
filters=16
size=3
stride=1
pad=1
activation=leaky

[maxpool]
size=2
stride=2

# (16, 320, 320)

Answer 1

Прямой путь - заглянуть в документацию по pytorch, где можно найти формулы, примененные в слоях.

Pytorch Conv2D Формулы для вычисления High и Width

Answer 2

Короче говоря, есть общая формула для расчета выходных яркости:

Вы можете найти объяснение в Руководство по арифметике рецептивного поля для сверточных нейронных сетей .

Кроме того, я хотел бы рекомендовать удивительную статью Руководство по арифметике свертки для глубокого обучения .

А это репо conv_arithmetic с анимацией свертки.