Как уменьшить размер .tflite, который я конвертировал из керас - PullRequest
0 голосов
/ 16 июня 2019

Я только что преобразовал Keras model .h5(resnet50 img classify), созданный в Google COLAB, в .tflite, но размер просто уменьшился, как 0.5 MB.

Это для мобильного приложения и размер должен уменьшиться больше.Это как 90 МБ и слишком большой для производства.

Платформа и распространение ОС = Linux-4.14.79 + -x86_64-with-Ubuntu-18.04-bionic

Версия TensorFlow = '1.14.0-rc1'

CUDA/ cuDNN версия = V10.0.130

Модель и память графического процессора = 11441 МБ |Tesla K80

from tensorflow import lite
converter = lite.TFLiteConverter.from_keras_model_file( '/gdrive/My Drive/Places/model.h5')
tfmodel = converter.convert()
open ("model.tflite" , "wb") .write(tfmodel)

Я ожидаю, что размер уменьшится, поскольку Firebase поддерживает до 40 МБ?Где я ошибаюсь или что мне делать?Я даже использовал pytorch и конвертировал до .pb из .pth, но не смог сделать

Как конвертировать из .pb в .tflite?

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...