Создайте файл TFLite с InceptionV3 и Food101

Question

Я пытаюсь создать модель TFLITE для использования на Android. Я использую InceptionV3Model с набором данных Food101. Я новичок в этом всем мире ML. Я нашел этот код от TensorFlow, который я пытаюсь использовать:

https://github.com/tensorflow/hub/blob/master/examples/image_retraining/retrain.py

Я не могу понять, куда добавить и что добавить для преобразования модели в TFLite.

Пожалуйста, объясните мне о контрольных точках и замороженных графиках, и помогите создать модель TFLite.

Спасибо.

Answer 1

Вы можете сначала экспортировать модель в save_model, установив --saved_model_dir, а затем используя TFLiteConverter.from_saved_model.

https://www.tensorflow.org/versions/r1.15/api_docs/python/tf/lite/TFLiteConverter#from_saved_model