конвертировать model.pb в model.tflite

Question

У меня есть сборка режима трансляции ML с кератами, у меня 50 checkoint_epoch.hdf5, а затем я нашел способ сохранить

весь режим в одном файле, который

model.save("model.h5")

и затем я нашел инструмент, который преобразовал этот файл в model.pb , теперь я должен изменить его на model.tflite , чтобы использовать его в мобильном приложении, которое ioni c v4 и angular

и как мне преобразовать model.pb в model.tflite

Answer 1

Самый простой способ - использовать API python.

В TF2.x для преобразования вашей модели из файла используйте:

converter = tf.lite.TFLiteConverter.from_saved_model(export_dir)
tflite_model = converter.convert()

Более подробную информацию можно найти здесь: https://www.tensorflow.org/lite/convert/python_api

В TF1.x конвертируйте ваш файл, используя:

graph_def_file = "/path/model.pb"
input_arrays = ["input"]
# Change the following to your application
output_arrays = ["MobilenetV1/Predictions/Softmax"]

converter = tf.lite.TFLiteConverter.from_frozen_graph(
  graph_def_file, input_arrays, output_arrays)
tflite_model = converter.convert()
open("converted_model.tflite", "wb").write(tflite_model)

Подробнее здесь: https://github.com/tensorflow/tensorflow/blob/master/tensorflow/lite/g3doc/r1/convert/python_api.md

Answer 2

См. https://www.tensorflow.org/lite/convert Существует python API и инструмент командной строки.