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

Question

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

Answer 1

发现这个更新的 python_api.md

我遇到了类似的问题并使用了这个：import tensorflow as 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

支持的操作列表可在此处获得。推荐的两个解决方案是：

1. 如此处所述添加自定义操作。
2. 仅在模型中使用 TFLite 当前支持的操作。