【发布时间】:2020-05-04 10:51:57
【问题描述】:
最近我使用 keras 来转移训练 mobilenet 模型。我重新配置了输出层以预测两个类。
然后我将保存的 .h5 文件转换为 .tflite 文件。根据 TensorFlow 的 ImageClassification 示例在 Android Studio 中运行一次:
https://github.com/tensorflow/examples/tree/master/lite/examples/image_classification
我发现预测的可信度存在细微差异。即
对于这张图片: [推理的图像][1]:https://i.stack.imgur.com/uSmLp.jpg
Tflite 模型置信度: [2.0677739E-5, 0.9999794]
使用 Keras 预测在 .h5 桌面上的推理: [5.100073E-4, 9.994900E-1]
如您所见,他们预测的大体方向相同。然而,我很好奇可能导致这种差异的原因是什么?
如果这不像一般解释那么简单,我将发布所有需要的代码。
桌面版本:
TensorFlow 2.1.0
Keras 2.3.1
Keras-applications 1.0.8
keras-preprocessing 1.1.0
安卓版本:
'org.tensorflow:tensorflow-lite:2.1.0'
'org.tensorflow:tensorflow-lite-gpu:0.0.0-nightly'
'org.tensorflow:tensorflow-lite-support:0.0.0-nightly'
谢谢!
【问题讨论】:
标签: android tensorflow keras tensorflow-lite