Pytorch adaptive_avg_pool2d алгоритм - PullRequest
0 голосов
/ 15 октября 2018

Кто-нибудь знает алгоритм для pytorch adaptive_avg_pool2d, например,

adaptive_avg_pool2d(image,[14,14])

, поэтому вопрос: я хочу сделать то же самое в нейронной сети keras, для любых входных данных, хочу получить 14 *14 выходных.Есть предложения?

1 Ответ

0 голосов
/ 03 мая 2019

Я не думаю, что это существует в Керасе.Вы можете получить размер вашего ввода и поделить на 14, чтобы получить желаемый размер пула.

Например, если ваши входы 28x28, вы можете использовать:

keras.layers.AveragePooling2D(pool_size=(2, 2), strides=None, padding='valid', data_format=None)

...