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

Question

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

Answer 1

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

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

Answer 2

找到此更新的python_api.md

我遇到了类似的问题，并使用了以下方法：     将tensorflow导入为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)

这种转换方法对我有用。