Модуль: tf.keras.applications.mobilenet
Модели MobileNet v1 для Keras.
MobileNet — это общая архитектура и может использоваться для множества задач. В зависимости от задачи, она может использовать различные размеры входного слоя и разные коэффициенты ширины. Это позволяет создавать модели с различной шириной, уменьшая количество умножений-сложений и тем самым снижая стоимость вычислений на мобильных устройствах.
MobileNets поддерживают любой размер входного изображения больше 32x32, причём большие размеры изображений обеспечивают лучшую производительность. Количество параметров и умножений-сложений можно изменить, используя параметр alpha, который увеличивает/уменьшает количество фильтров в каждом слое. Изменяя размер изображения и параметр alpha, можно построить все 16 моделей из статьи, с предоставленными весами ImageNet.
В статье показана производительность MobileNets с использованием значений alpha 1.0 (также называемого 100% MobileNet), 0.75, 0.5 и 0.25. Для каждого из этих значений alpha предоставлены веса для 4 различных размеров входных изображений (224, 192, 160, 128).
В следующей таблице описаны размер и точность 100% MobileNet
на размере 224x224:
Коэффициент ширины (alpha) | Точность ImageNet | Умножения-сложения (M) | Параметры (M)
| 1.0 MobileNet-224 | 70,6 % | 529 | 4,2 | | 0,75 MobileNet-224 | 68,4 % | 325 | 2,6 | | 0,50 MobileNet-224 | 63,7 % | 149 | 1,3 |
| 0,25 MobileNet-224 | 50,6 % | 41 | 0,5 |
В следующей таблице описана производительность
100% MobileNet на различных размерах входных данных:
Resolution | ImageNet Acc | Multiply-Adds (M) | Params (M)
| 1.0 MobileNet-224 | 70,6 % | 569 | 4,2 | | 1.0 MobileNet-192 | 69,1 % | 418 | 4,2 | | 1.0 MobileNet-160 | 67,2 % | 290 | 4,2 |
| 1,0 MobileNet-128 | 64,4 % | 186 | 4,2 |
Ссылка:
Функции
MobileNet(...): Инициализирует архитектуру MobileNet.
decode_predictions(...): Декодирует предсказание модели ImageNet.
preprocess_input(...): Предварительно обрабатывает тензор или массив NumPy, кодирующий пакет изображений.
© 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/versions/r2.9/api_docs/python/tf/keras/applications/mobilenet