Скомпилируйте AutoML TFLite в Edge TPU

Я использовал Vision AutoML для обучения специальной модели. Мой набор данных содержит панорамные изображения (3000x1500 пикселей). Эта модель отлично работает на tensorflow.lite (2.5.0). Но я хочу загрузить его на Coral Edge TPU.

Когда я использую edge_compiler, выдает эту ошибку:

Компилятор Edge TPU версии 15.0.340273435 loc (reshape_output): ошибка: параметры квантования 'tfl.reshape' нарушают одно и то же ограничение масштаба:! Quant.uniform ‹u8: f32, 0.43928399682044983: 111› vs.! Quant.uniform ‹u8: f32, 0.4234168529510498: 111 ›

Внутренняя ошибка компилятора. Прерывание!

Любые идеи ?


person Bertrand Rodrigues    schedule 08.04.2021    source источник


Ответы (1)


Просто нужно перейти на edgetpu_compiler 14.1

person Bertrand Rodrigues    schedule 08.04.2021