Softmax Tensorflow Lite 行为不正常

Question

我制作了一个简单的卷积 Tensorflow 模型，它使用 softmax 进行推理。当我在 python 中运行模型并为模型提供图像时，一切都会相应地工作。但是，当我使用 tflite 转换模型并从 android 运行它时，输出是一个全 0 的列表。不知何故，softmax 不起作用。当我删除 softmax 函数时，python 和 android 中的输出是相同的，我可以通过在 Java 中实现 softmax 函数来修复它，但应该有更好的方法吧？我试过改变 softmax 函数的轴，但它在 Java 中只返回 0。

谢谢！

y_test = tf.nn.softmax(test_network[0])

Answer 1

大概您使用的是量化图。 TFLite 以浮点计算 softmax，然后将其乘以 256 并存储为 uint8。由于大多数值非常小，因此在乘法和四舍五入到最接近的整数后，它们将为零。不过排名应该是一样的，如果要显示softmax，可以除以256，显示前几个非零值。