Модуль: 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
при размере 224 x 224:
Коэффициент ширины (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 % | 529 | 4.2 | | 1.0 MobileNet-192 | 69.1 % | 529 | 4.2 | | 1.0 MobileNet-160 | 67.2 % | 529 | 4.2 |
| 1.0 MobileNet-128 | 64.4 % | 529 | 4.2 |
Статья:
Функции
MobileNet(...): Инициализирует архитектуру MobileNet.
decode_predictions(...): Декодирует предсказание модели ImageNet.
preprocess_input(...): Предварительно обрабатывает тензор или массив NumPy, кодирующий пакет изображений.
© 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/mobilenet