Android：将Tensorflow 2.0 SavedModel转换为具有置信度值的TFLite问题

Question

您好，使用此源代码https://gist.github.com/mypapit/e3b26787c95caf840e5c16a79327d443在TFHub MobileNet模块（SavedModel TF2.0）上使用转移学习技术后，我已经将SavedModel文件转换为TFLite文件，并且我尝试在用于诗人示例Android应用程序的Tensorflow上运行它

image 1

生成的Android应用程序似乎能够正确分类我经过重新训练的课程。但是，准确性相差甚远（就像荒唐可笑！），您可以参考屏幕截图。

通常，精度应在0.000到1.000的范围内。但是使用我转换的TFLite模型，精度范围从-400到500.00 ++千差万别]

仅供参考，我已经修改IMAGE_MEAN和IMAGE_STD值（255f，0f）和（127.5f，127.5f），但无济于事。

有人可以帮我吗？

我用来测试模型的标准TensorFlow for Poets Android源代码在这里：https://gist.github.com/mypapit/f7a9b54ee502f02ca72da3f972d25fb9

转换后的TFLite文件在这里：https://1drv.ms/u/s!AmVw1Hsqu0-CguVlKyCNE0W-NzODEg?e=LkjBXl其标签图在这里：https://gist.github.com/mypapit/56845dde0c47e21d0e18ec86d25a3ff2

[我注意到这仅在将Tensorflow 2.x与tfhub模块（TF2.0 SavedModel）一起使用时发生，而当我与Tensorflow 1.14一起使用tfhub模块时没有发生

我已经不知所措，有人可以帮我吗？ ：（

Answer 1

似乎您缺少最后一个softmax图层，该图层使所有标签输出的总和为1。

https://medium.com/data-science-bootcamp/understand-the-softmax-function-in-minutes-f3a59641e86d