代码之家 › 专栏 › 技术社区 › william007

Tensorflow Lite不支持tf.less-如何处理它

tensorflow-lite tensorflow

1

william007 · 技术社区 · 8 年前

我使用TensorFlow来训练我的模型,然后将其转换为TensorFlow Lite模型,但是运行时它显示 tf.less , tf.greaterthan 不支持。我该怎么办?

2 回复 | 直到 8 年前

1

0

Nupur Garg 8 年前

支持的操作列表可用 here . 推荐的两种解决方案是:

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

2

0

PixelPioneer 8 年前

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

这种转换方法对我有效。