TF 2.0图像分类中的.tflite模型带来麻烦

时间:2019-09-03 03:26:13

标签: tensorflow

我刚刚使用本指南https://www.tensorflow.org/beta/tutorials/images/hub_with_keras构建了 TF 2.0 RC0图像分类模型,然后将该模型转换为.tflite

converter = tf.lite.TFLiteConverter.from_keras_model(model)
tflite_model = converter.convert()
open("converted_model.tflite", "wb").write(tflite_model)

当我尝试将.tflite与https://github.com/tensorflow/tensorflow/blob/master/tensorflow/lite/examples/python/label_image.py的label_image脚本一起使用时,无法正确分类图像。该分类可与model.predict配合使用(即在转换为.tflite之前)

知道我可能做错了什么或如何解决此问题吗?

0 个答案:

没有答案