tf.keras.applications.DenseNet201
| Просмотреть исходный код на GitHub |
Инициализирует архитектуру Densenet201.
tf.keras.applications.DenseNet201(
include_top=True, weights='imagenet', input_tensor=None, input_shape=None,
pooling=None, classes=1000
)
Статья по ссылке:
- Плотновъязанные сверточные сети (Премия за лучшую статью CVPR 2017)
При необходимости загружает предварительно обученные веса на наборе данных ImageNet. Обратите внимание, что соглашение о формате данных, используемое моделью, соответствует настройкам Keras, указанным в ~/.keras/keras.json.
| Аргументы | |
|---|---|
include_top | включить ли полносвязный слой в верхней части сети. |
weights | одно из значений None (случайная инициализация), 'imagenet' (предварительная подготовка на ImageNet) или путь к файлу весов для загрузки. |
input_tensor | необязательный тензор Keras (например, результат работы layers.Input()) для использования в качестве входного изображения для модели. |
input_shape | необязательная кортеж размера, который нужно указывать только если include_top равно False (в противном случае размер входа должен быть (224, 224, 3) (с форматом данных 'channels_last' ) или (3, 224, 224) (с форматом данных 'channels_first' ). Он должен иметь ровно 3 канала, а ширина и высота не должны быть меньше 32. Например, (200, 200, 3) будет допустимым значением. |
pooling | Необязательный режим пулинга для извлечения признаков, когда include_top равно False.
|
classes | необязательное число классов для классификации изображений, которое нужно указывать только если include_top равно True и если аргумент weights не указан. |
| Возвращаемое значение | |
|---|---|
| Экземпляр модели Keras. |
© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.3/api_docs/python/tf/keras/applications/DenseNet201