tf.less не поддерживается в tenorflow lite - как с этим бороться - PullRequest
0 голосов
/ 22 мая 2018

Я использую tenorflow для обучения своей модели, затем преобразовываю ее в тензор потока облегченной модели, но при запуске показывает, что tf.less, tf.greaterthan не поддерживаются.Как мне обойти это?

Ответы [ 2 ]

0 голосов
/ 26 сентября 2018

Нашел это обновленное python_api.md

Я столкнулся с похожими проблемами и использовал это: импортировать тензор потока как tf

converter = tf.contrib.lite.TocoConverter.from_keras_model_file("keras_model.h5")
tflite_model = converter.convert()
open("converted_model.tflite", "wb").write(tflite_model)

Этот метод преобразования работал мне.

0 голосов
/ 27 июля 2018

Список поддерживаемых операций доступен здесь .Два рекомендуемых решения:

  1. Добавьте пользовательскую операцию, как описано здесь .
  2. Используйте только операции, которые в настоящее время поддерживаются TFLite в вашей модели.
...