Преобразование файла retrained_graph.pb в модель tflite - PullRequest
Новая
       13

Преобразование файла retrained_graph.pb в модель tflite

0 голосов
/ 10 июля 2019

Я не могу преобразовать модель файла retrained_graph.pb (inception-V3), полученную в результате переобучения, в модель .tflite для приложения Android.

Код для записи файла retrained_graph.pb: 

    output_graph_def = graph_util.convert_variables_to_constants(sess, graph.as_graph_def(), [FINAL_TENSOR_NAME])
    with gfile.FastGFile(graph_file_name, 'wb') as f:
        f.write(output_graph_def.SerializeToString())

При преобразовании в tflite возникает ошибка: RuntimeError: MetaGraphDef, связанный с тегами {'serve'}, не найден вSavedModel.

Код: 

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

Итак, кто-нибудь, пожалуйста, помогите мне с этой проблемой, чтобы я мог легко преобразовать файл .pb в файл .tflite.

1 Ответ

0 голосов
/ 13 июля 2019

Я могу преобразовать в файл модели retrained_graph.pb, полученный с помощью метода обучения обучению, в файл tflite, используя следующую команду:

toco --graph_def_file = / save_model.pb --output_file = C: /Users/sunil.gc/tf-python/src/saved_model.tflite --input_format = TENSORFLOW_GRAPHDEF --output_format = TFLITE --input_999 3 --input_array = Mul --output_array = final_result --inference_type = FLOAT --input_type = FLOAT

...