Я использую tf2.0 и API обнаружения объектов. У меня есть предварительно обученная модель ssd_mobilenet_v2_coco, и я хочу преобразовать ее в тензор потока lite (и выполнить квантование float16).
В папке модели у меня есть контрольная точка, frozen_inference_graph.pb, model.ckpt.data-00000-of-00001, model.ckpt.index, model.ckpt.meta, pipe.config и Папка save_model.
Я использовал следующую команду для получения tflite.pb python3 export_tflite_ssd_graph.py --output_directory = / home / tflite --pipeline_config_path = / home / exportedLP1 / pipeline.config --trained_checkpoint_pref /home/exportedLP1/model.ckpt
Затем я использовал следующий код python для преобразования tflite.pb:
frozen_model_path = '/home/tflite/tflite_graph.pb'
input_arrays=["image_tensor"]
output_arrays=["detection_boxes","detection_scores","num_detections","detection_classes"]
input_tensor={"image_tensor":[1,300,300,3]}
converter = tf.compat.v1.lite.TFLiteConverter.from_frozen_graph(frozen_model_path, input_arrays, output_arrays, input_tensor)
tflite_model = converter.convert()
Последний шаг tflite = converter.convert () вызывает " abort (core dumped) '. В соответствии с сообщениями, которые я читаю в Интернете, я могу без проблем конвертировать в tflite.
Текущая тема 0x00007efbff6f1740 (самый последний вызов - первый):
File "/usr/local/lib/python3.6/dist-packages/tensorflow_core/lite/toco/python/toco_from_protos.py", line 56 in execute
File "/usr/local/lib/python3.6/dist-packages/absl/app.py", line 250 in _run_main
File "/usr/local/lib/python3.6/dist-packages/absl/app.py", line 299 in run
File "/usr/local/lib/python3.6/dist-packages/tensorflow_core/python/platform/app.py", line 40 in run
File "/usr/local/lib/python3.6/dist-packages/tensorflow_core/lite/toco/python/toco_from_protos.py", line 93 in main
File "/usr/local/bin/toco_from_protos", line 8 in
Aborted (core dumped)