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

Question

Я создал модель обнаружения объектов, используя Pytorch, а затем преобразовал из .pth в .onnx, а затем .pb, но теперь мне нужно преобразовать ее в .tflite для приложения для Android!Как это сделать?Это мой первый раз.

input_arrays  = [64, 3, 224, 224] 

output_arrays = ? 

для бинарной классификации.

Я сделал это из pytorch, но все, что я вижу, было из керас или Tensorflow ...

Это код, который я использовал для преобразования из .pb в .tflie

converter = lite.TFLiteConverter.from_frozen_graph(
  model/model.pb , input_arrays, output arrays )
tflite_model = converter.convert()
open("converted_model.tflite", "wb").write(tflite_model)

!tflite_convert \
   --output_file= model/model.tflite \
   --graph_def_file= model/model.pb \
   --input_arrays= input_arrays \
   -- output_arrays= output_arrays

Я думаю, что это как-то связано с входными массивами и выходными массивами, но не уверен в этом,graph_def_file должен хранить model.pb?

Answer 1

Нет необходимости указывать массив ввода и вывода при использовании следующего кода:

import tensorflow as tf

converter = tf.lite.TFLiteConverter.from_saved_model(saved_model_dir)
tflite_model = converter.convert()
open("converted_model.tflite", "wb").write(tflite_model)

Попробуйте это.