edgetpu_compiler: как объединить две модели tflite - PullRequest
Новая
       11

edgetpu_compiler: как объединить две модели tflite

0 голосов
/ 22 октября 2019

Я пытаюсь использовать Google Coral edgetpu_compiler для объединения двух существующих моделей tflite в одну модель , следуя указаниям Google . Я использую две из предварительно скомпилированных моделей Google . Ошибка указывает на то, что модели уже скомпилированы для устройства Coral. Эти модели на самом деле уже скомпилированы для Edge TPU, но я пытаюсь объединить две модели. Я делаю что-то не так или объединение моделей Edge TPU не поддерживается?

Вот команда, которую я запускаю, и вывод: 

$ edgetpu_compiler \
    mobilenet_ssd_v2_coco_quant_postprocess_edgetpu.tflite \
    mobilenet_ssd_v2_face_quant_postprocess_edgetpu.tflite
Edge TPU Compiler version 2.0.267685300
Invalid model: mobilenet_ssd_v2_coco_quant_postprocess_edgetpu.tflite
Model already compiled for Edge TPU

Я запускаю это на версии ОС Coral Board: Mendel GNU/Linux 3 (Chef).

Любое руководство приветствуется.

Спасибо, Джон

1 Ответ

1 голос
/ 22 октября 2019

Модели, которые вы пытаетесь объединить, уже скомпилированы для edgetpu и не могут быть скомпилированы снова.

Чтобы снова скомпилировать обе модели, вы можете загрузить «Все файлы моделей» «MobileNet SSD v2 (COCO)» и «MobileNet SSD v2 (Faces)» из https://coral.withgoogle.com/models/. После извлечения этихсжатые файлы, вы найдете файлы tflite_graph.pb для обеих моделей. Вам нужно будет конвертировать эти файлы .pb в .tflite (версия CPU) . После получения версии CPU для файлов .tflite вы сможете собрать две модели вместе.

Подробные требования к модели см. По адресу: https://coral.withgoogle.com/docs/edgetpu/models-intro/#compatibility-overview

Информацию о совместной компиляции см. По адресу: https://coral.withgoogle.com/docs/edgetpu/compiler/#co-compiling-multiple-models. Обратите внимание, что совместная компиляция n моделейвыпускаю n моделей, а не только одну. Преимущество заключается в том, что данные параметров поддержки компилятора кэшируются вместе в ОЗУ ТПУ

...