Как найти формы активаций в разных слоях предварительно обученной модели InceptionResNetV2 в Keras - Tensorflow 2.0 - PullRequest
1 голос
/ 09 октября 2019

У меня загружена модель inceptionResNetV2 Keras

base_model = tf.keras.applications.inception_resnet_v2.InceptionResNetV2(include_top=False, weights='imagenet')

Я хочу найти формы активаций, выдаваемых разными слоями - при условии стандартного размера ввода (299x299).

Моя конечная цель - принять обоснованное решение о том, какую часть предварительно обученной модели оставить неподготовленным (используя также другие критерии).

Я пытался:

base_model.summary()

Что возвращает:

enter image description here

Аналогично, когда я пытаюсь:

enter image description here

Другими словами, я получаю глубину (количество фильтров) тензора активации, но не ширину / высоту.

Что мне нужно сделать, чтобы найти форму активаций после ввода изображения (299x299)в сеть?

1 Ответ

1 голос
/ 09 октября 2019

Вы можете поместить input_shape в функцию с помощью

base_model = tf.keras.applications.inception_resnet_v2.InceptionResNetV2(include_top=False, weights='imagenet', input_shape=(299, 299, 3))

Но это вызовет ошибку, если входные изображения не 299 * 299, поэтому лучше использовать его, только когда вы хотите знать форму.

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...