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 требуется определённый тип предварительной обработки входных данных. Для NASNet перед передачей входных данных модели вызовите для них keras.applications.nasnet.preprocess_input.
| Аргументы | |
|---|---|
input_shape | Необязательный кортеж формы, который указывается только если include_top имеет значение False (в противном случае форма входных данных для NASNetMobile должна быть (224, 224, 3)). Он должен содержать ровно 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