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

Question

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

标签： tensorflowtensorflow-lite

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)

这种转换方法对我有用。