Ошибка преобразования .pb модели tenorflow в .mlmodel

Question

Я пытаюсь преобразовать график тензорного потока ( .pb файл ) в .mlmodel

import tfcoreml
coreml_model = tfcoreml.convert(tf_model_path='optimized_model.pb', mlmodel_path='FaceImages.mlmodel', output_feature_names=['final_result'], input_name_shape_dict={'ResizeBilinear': {'images': None, 'size': {None, None}}}, minimum_ios_deployment_target='13')

, но получаю следующую ошибку:

/ usr / local / lib / python3 .6 / dist-packages / coremltools / конвертеры / nnssa / внешний интерфейс / тензор потока / graphdef_to_ssa.py в load_tf_graph (graph_file)

 21     with tf.io.gfile.GFile(graph_file, "rb") as f:
 22         graph_def = tf.compat.v1.GraphDef()

---> 23 graph_def.ParseFromString(f.read())

 24 
 25     # Then, we import the graph_def into a new Graph and returns it

DecodeError: Ошибка разбора сообщения

Может ли кто-нибудь помочь с этим PLS?

Вот проект Colab , где я приложил модель тензор потока и связанный код для преобразования

https://colab.research.google.com/drive/1S7nf7pnX15UuswFZaTih5pHhfDFwG5Xa

Answer 1

Проверяли ли вы, что используемая вами версия Tensorflow совместима с этими библиотеками? Это всего лишь предположение, но вы можете попробовать запустить

!pip install tensorflow --upgrade

в верхней части ноутбука, чтобы посмотреть, решит ли это проблему.