Почему не работает Coral USB при запуске собственной модели tflite?

Установка: Raspberry PI 4b 4GB с ускорителем Coral USB.

Я использую эту программу обнаружения официального Coral Веб-сайт USB.

Обнаружение запуска с помощью mobilenet_ssd_v2_face_quant_postprocess_edgetpu.tflite работает со скоростью 30 кадров в секунду. И светодиод на Coral USB мигает, , что означает, что Edge TPU работает.

python3 detect.py --model=../all_models/mobilenet_ssd_v2_face_quant_postprocess_edgetpu.tflite --labels=../all_models/coco_labels.txt

И на выходе получается примерно такой поток:

96% person Inference: 11.99ms FPS:22.9

Запуск mymodel.tflite работает со скоростью 3 кадра в секунду. Светодиод на коралловом USB горит постоянно, это означает, что Coral USB инициализирован, но не работает.

python3 detect.py --model=../all_models/mymodel.tflite --labels=../all_models/dict.txt

И вывод - это поток чего-то вроде tis:

85% S_13 Inference: 290.25ms FPS:3.2

Примеры с gstreamer, pygame и raspicam в репозитории GitHub дают тот же результат (3 кадра в секунду) и постоянный свет, если я запускаю свою собственную модель, но все они работают со скоростью 30 кадров в секунду с мобильной сетью, и светодиод мигает во всех трех случаях.

Я считаю, что проблема связана с тем, что Coral USB не работает, когда Raspberry выполняет мою собственную модель, в которой есть эти спецификации

Другое

Мои ярлыки имеют тот же формат, что и coco_labels.txt, и обнаружение, даже если оно медленное, происходит. Таким образом, ярлыки, похоже, не проблема.


person fbu_94    schedule 18.06.2020    source источник


Ответы (1)


Мне кажется, что вы не пропустили свою модель через edgetpu_compiler.

edgetpu_compiler -s mymodel.tflite

Если это не удается, проверьте, соответствует ли ваша пользовательская модель требованиям здесь: https://coral.ai/docs/edgetpu/models-intro/#model-requirements

person Nam Vu    schedule 19.06.2020
comment
Это помогло скомпилировать модель, спасибо. Он работает быстрее, но теперь ничего не обнаруживает. Это результат компиляции: Модель успешно скомпилирована, но не все операции поддерживаются Edge TPU. Вместо этого часть модели будет работать на процессоре, который работает медленнее. Если возможно, подумайте об обновлении вашей модели, чтобы использовать только операции, поддерживаемые Edge TPU. Для получения подробной информации посетите g.co/coral/model-reqs. Количество операций, которые будут выполняться на Edge TPU: 132 Количество операций, которые будут выполняться на ЦП: 162 - person fbu_94; 24.06.2020
comment
О, это случайно не модель обнаружения объектов AutoML? - person Nam Vu; 24.06.2020
comment
Он был создан с помощью Google Vision и экспортирован, выбрав опцию tflite. - person fbu_94; 25.06.2020
comment
Итак, чтобы ответить на вопрос: да, это обнаружение объекта AutoML. Что мне не хватает? - person fbu_94; 25.06.2020
comment
Ах, ладно, извините за это, отправьте свою модель по адресу [email protected] и укажите эту проблему. Есть некоторая несовместимость с выпущенным компилятором и AutoML, если вы пришлете его нам, я скомпилирую для вас. Все это будет исправлено после следующего выпуска компилятора! - person Nam Vu; 25.06.2020
comment
Это очень добро. Спасибо! Я сделаю это. - person fbu_94; 25.06.2020