tensorflow lite 不支持 tf.less - 如何解决它

Question

我使用 tensorflow 训练我的模型，随后将其转换为 tensorflow lite 模型，但运行时显示不支持 tf.less、tf.greaterthan。我应该如何解决这个问题？

Answer 1

发现此更新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)

这种转换方法对我有用。

Answer 2

支持的操作列表可用here。推荐的两个解决方案是：

1. 按照here 的描述添加自定义操作。
2. 仅在您的模型中使用 TFLite 当前支持的操作。