tf.keras.applications.NASNetMobile
Инициализирует модель Mobile NASNet в режиме ImageNet.
tf.keras.applications.NASNetMobile(
input_shape=None,
include_top=True,
weights='imagenet',
input_tensor=None,
pooling=None,
classes=1000,
classifier_activation='softmax'
)
Ссылка:
При необходимости загружает предварительно обученные веса на ImageNet. Обратите внимание, что соглашение о формате данных, используемое моделью, соответствует соглашению, указанному в вашей конфигурации Keras по адресу ~/.keras/keras.json.
Примечание: каждая Keras Application ожидает определённого вида предобработки входных данных. Для NASNet вызовите keras.applications.nasnet.preprocess_input для ваших входных данных перед передачей их в модель.
| Аргументы | |
|---|---|
input_shape | Необязательная кортеж формы, который следует указать только если include_top равно False (в противном случае форма входных данных должна быть (224, 224, 3) для NASNetMobile. Она должна иметь ровно 3 канала, а ширина и высота не должны быть меньше 32. Например, (224, 224, 3) — это допустимое значение. |
include_top | Включать ли полносвязный слой в верхней части сети. |
weights | None (случайная инициализация) или imagenet (веса ImageNet). Для загрузки imagenet весов, input_shape должна быть (224, 224, 3) |
input_tensor | Необязательный тензор Keras (т.е. результат работы layers.Input()) для использования в качестве входных данных изображения для модели. |
pooling | Необязательный режим пулинга для извлечения признаков, когда include_top равен False.
|
classes | Необязательное количество классов для классификации изображений, которое следует указывать только если include_top равно True, и если аргумент weights не указан. |
classifier_activation | Функция активации для использования на верхнем слое. Функция может быть str или вызываемой функцией. Она игнорируется, если include_top=True. Установите classifier_activation=None, чтобы получить логарифмы вероятностей для верхнего слоя. При загрузке предварительно обученных весов, classifier_activation может быть только None или "softmax". |
| Возвращаемое значение | |
|---|---|
| Экземпляр модели Keras. |
© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/api_docs/python/tf/keras/applications/NASNetMobile